Spaces:
Running
on
Zero
Running
on
Zero
parokshsaxena
commited on
Commit
•
4209785
1
Parent(s):
6b06100
putting enhanced garment net in app.py
Browse files
app.py
CHANGED
|
@@ -53,7 +53,8 @@ unet = UNet2DConditionModel.from_pretrained(
|
|
| 53 |
)
|
| 54 |
unet.requires_grad_(False)
|
| 55 |
|
| 56 |
-
enhancedGarmentNet = EnhancedGarmentNetWithTimestep(
|
|
|
|
| 57 |
|
| 58 |
tokenizer_one = AutoTokenizer.from_pretrained(
|
| 59 |
base_path,
|
|
|
|
| 53 |
)
|
| 54 |
unet.requires_grad_(False)
|
| 55 |
|
| 56 |
+
enhancedGarmentNet = EnhancedGarmentNetWithTimestep()
|
| 57 |
+
enhancedGarmentNet.to(dtype=torch.float16)
|
| 58 |
|
| 59 |
tokenizer_one = AutoTokenizer.from_pretrained(
|
| 60 |
base_path,
|