SVD-XT-1.1

Runtime error

Manjushri commited on Feb 13, 2024

Commit

cdd397f

verified ·

1 Parent(s): f2ba17b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -30,7 +30,7 @@ def sample(
     version: str = "svd_xt",
     cond_aug: float = 0.02,
     decoding_t: int = 3,  # Number of frames decoded at a time! This eats most VRAM. Reduce if necessary.
-    device: str = "cpu",
     output_folder: str = "outputs",):
     if image.mode == "RGBA":

     version: str = "svd_xt",
     cond_aug: float = 0.02,
     decoding_t: int = 3,  # Number of frames decoded at a time! This eats most VRAM. Reduce if necessary.
+    device: str = "cuda",
     output_folder: str = "outputs",):
     if image.mode == "RGBA":