Spaces:

John6666
/

votepurchase-multiple-model

Running on Zero

John6666 commited on Sep 20

Commit

77058ee

•

1 Parent(s): 15a474c

Upload dc.py

Files changed (1) hide show

dc.py CHANGED Viewed

@@ -320,7 +320,7 @@ class GuiSD:
     def infer_short(self, model, pipe_params, progress=gr.Progress(track_tqdm=True)):
         progress(0, desc="Start inference...")
-        images, image_list, metadata = model(**pipe_params)
         progress(1, desc="Inference completed.")
         if not isinstance(images, list): images = [images]
         img = []

     def infer_short(self, model, pipe_params, progress=gr.Progress(track_tqdm=True)):
         progress(0, desc="Start inference...")
+        images, seed, image_list, metadata = model(**pipe_params)
         progress(1, desc="Inference completed.")
         if not isinstance(images, list): images = [images]
         img = []