Spaces:
Sleeping
Sleeping
Model Path Changes
Browse files
app.py
CHANGED
|
@@ -48,6 +48,7 @@ unet = UNet2DConditionModel.from_pretrained(
|
|
| 48 |
subfolder="unet",
|
| 49 |
torch_dtype=torch.float16,
|
| 50 |
)
|
|
|
|
| 51 |
unet.requires_grad_(False)
|
| 52 |
tokenizer_one = AutoTokenizer.from_pretrained(
|
| 53 |
base_path,
|
|
|
|
| 48 |
subfolder="unet",
|
| 49 |
torch_dtype=torch.float16,
|
| 50 |
)
|
| 51 |
+
|
| 52 |
unet.requires_grad_(False)
|
| 53 |
tokenizer_one = AutoTokenizer.from_pretrained(
|
| 54 |
base_path,
|