Manju-Dream-Booth-GPU

Sleeping

Manjushri commited on Feb 12

Commit

1152269

•

1 Parent(s): 3feebcb

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":
@@ -101,4 +101,4 @@ with gr.Blocks() as demo:
 if __name__ == "__main__":
     demo.queue(max_size=20, api_open=False)
-    demo.launch(share=True, show_api=False)

     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":
 if __name__ == "__main__":
     demo.queue(max_size=20, api_open=False)
+    demo.launch(show_api=False)