Przemyslaw2115 commited on
Commit
f1ab259
·
verified ·
1 Parent(s): 1bb8f75

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
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")