Merge pull request #3534 from vladmandic/dev

refresh master
This commit is contained in:
Vladimir Mandic
2024-10-29 15:35:58 -04:00
committed by GitHub
2 changed files with 2 additions and 1 deletions
+1 -1
View File
@@ -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:
+1
View File
@@ -61,6 +61,7 @@ torchdiffeq
dctorch
scikit-image
seam-carving
sentencepiece
# block
torch!=2.5.0