From d205b5cf3e364bd81526e097cfe4d8834ee27944 Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Tue, 29 Oct 2024 15:31:50 -0400 Subject: [PATCH] update Signed-off-by: Vladimir Mandic --- modules/sd_models.py | 2 +- requirements.txt | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/modules/sd_models.py b/modules/sd_models.py index 734f4ab57..dc0679c58 100644 --- 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 index 208cfba3a..d376fe320 100644 --- a/requirements.txt +++ b/requirements.txt @@ -61,6 +61,7 @@ torchdiffeq dctorch scikit-image seam-carving +sentencepiece # block torch!=2.5.0