Upload folder using huggingface_hub
Browse files- .idea/workspace.xml +1 -1
- app.py +2 -2
.idea/workspace.xml
CHANGED
|
@@ -54,7 +54,7 @@
|
|
| 54 |
<option name="number" value="Default" />
|
| 55 |
<option name="presentableId" value="Default" />
|
| 56 |
<updated>1760056680813</updated>
|
| 57 |
-
<workItem from="1760056681869" duration="
|
| 58 |
</task>
|
| 59 |
<servers />
|
| 60 |
</component>
|
|
|
|
| 54 |
<option name="number" value="Default" />
|
| 55 |
<option name="presentableId" value="Default" />
|
| 56 |
<updated>1760056680813</updated>
|
| 57 |
+
<workItem from="1760056681869" duration="4675000" />
|
| 58 |
</task>
|
| 59 |
<servers />
|
| 60 |
</component>
|
app.py
CHANGED
|
@@ -61,9 +61,9 @@ os.remove('checkpoints/vae.pth')
|
|
| 61 |
|
| 62 |
|
| 63 |
if torch.cuda.is_available():
|
| 64 |
-
|
| 65 |
else:
|
| 66 |
-
|
| 67 |
|
| 68 |
|
| 69 |
def fig_to_image(fig):
|
|
|
|
| 61 |
|
| 62 |
|
| 63 |
if torch.cuda.is_available():
|
| 64 |
+
model = model.to(torch.bfloat16).cuda()
|
| 65 |
else:
|
| 66 |
+
model = model.to(torch.float32)
|
| 67 |
|
| 68 |
|
| 69 |
def fig_to_image(fig):
|