Spaces:
Running
Running
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")
|