Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -23,6 +23,11 @@ import tempfile
|
|
| 23 |
import cv2
|
| 24 |
import os
|
| 25 |
import requests
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 26 |
|
| 27 |
snapshot_download(repo_id="LiuZichen/MagicQuill-models", repo_type="model", local_dir="models")
|
| 28 |
# HF_TOKEN = os.environ.get("HF_TOKEN")
|
|
|
|
| 23 |
import cv2
|
| 24 |
import os
|
| 25 |
import requests
|
| 26 |
+
import torch.nn
|
| 27 |
+
|
| 28 |
+
# Hack: wymuszamy float32 zamiast half()
|
| 29 |
+
torch.set_default_dtype(torch.float32)
|
| 30 |
+
torch.nn.Module.half = lambda self: self.float()
|
| 31 |
|
| 32 |
snapshot_download(repo_id="LiuZichen/MagicQuill-models", repo_type="model", local_dir="models")
|
| 33 |
# HF_TOKEN = os.environ.get("HF_TOKEN")
|