Spaces:

John6666
/

votepurchase-multiple-model

Running on Zero

John6666 commited on Sep 5, 2024

Commit

9ee7707

verified ·

1 Parent(s): 27218a1

Upload dc.py

Files changed (1) hide show

dc.py CHANGED Viewed

@@ -314,7 +314,7 @@ class GuiSD:
             vae_model=None,
             type_model_precision=torch.float16,
             retain_task_model_in_cache=False,
-            device="cpu",
         )
     def infer_short(self, model, pipe_params, progress=gr.Progress(track_tqdm=True)):

             vae_model=None,
             type_model_precision=torch.float16,
             retain_task_model_in_cache=False,
+            #device="cpu",
         )
     def infer_short(self, model, pipe_params, progress=gr.Progress(track_tqdm=True)):