Spaces:
Running
on
Zero
Running
on
Zero
safe model tensor updated
Browse files
app.py
CHANGED
@@ -48,7 +48,6 @@ unet = UNet2DConditionModel.from_pretrained(
|
|
48 |
subfolder="unet",
|
49 |
torch_dtype=torch.float16,
|
50 |
)
|
51 |
-
|
52 |
unet.requires_grad_(False)
|
53 |
tokenizer_one = AutoTokenizer.from_pretrained(
|
54 |
base_path,
|
|
|
48 |
subfolder="unet",
|
49 |
torch_dtype=torch.float16,
|
50 |
)
|
|
|
51 |
unet.requires_grad_(False)
|
52 |
tokenizer_one = AutoTokenizer.from_pretrained(
|
53 |
base_path,
|