Merge pull request #3534 from vladmandic/dev

refresh master
vladmandic · Oct 29, 2024 · 41f9a35 · 41f9a35
2 parents 99623da + d205b5c
commit 41f9a35
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/modules/sd_models.py b/modules/sd_models.py
@@ -477,7 +477,7 @@ def move_model(model, device=None, force=False):
     try:
         t0 = time.time()
         try:
-            model.to(device, non_blocking=True)
+            model.to(device)
             if hasattr(model, "prior_pipe"):
                 model.prior_pipe.to(device)
         except Exception as e0:

diff --git a/requirements.txt b/requirements.txt
@@ -61,6 +61,7 @@ torchdiffeq
 dctorch
 scikit-image
 seam-carving
+sentencepiece
 
 # block
 torch!=2.5.0