mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 09:14:35 +02:00
@@ -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:
|
||||
|
||||
@@ -61,6 +61,7 @@ torchdiffeq
|
||||
dctorch
|
||||
scikit-image
|
||||
seam-carving
|
||||
sentencepiece
|
||||
|
||||
# block
|
||||
torch!=2.5.0
|
||||
|
||||
Reference in New Issue
Block a user