add seed everything

app-huggingface.py

@@ -53,6 +53,16 @@ VAE_PATH = './models/VAE'
 DreamBooth_LoRA_PATH = './models/DreamBooth_LoRA'
 
 
+def seed_everything(seed):
+    import random
+
+    import numpy as np
+    torch.manual_seed(seed)
+    torch.cuda.manual_seed_all(seed)
+    np.random.seed(seed % (2**32))
+    random.seed(seed)
+
+
 if not LOCAL_DEBUG:
     CACHE_PATH = './models'
 
@@ -235,12 +245,12 @@ class AnimateController:
                 gr.Info('Use exmaple image for fast try!')
                 init_img = np.array(Image.open(example_img))
 
-        init_img_bk = copy.deepcopy(init_img)
         if seed_textbox != -1 and seed_textbox != "":
             torch.manual_seed(int(seed_textbox))
         else:
-            torch.seed()
-        seed = torch.initial_seed()
+            seed = torch.initial_seed()
+
+        seed_everything(seed)
 
         pipeline = self.pipeline_dict[style]
         init_img, h, w = preprocess_img(init_img)
@@ -278,7 +288,8 @@ class AnimateController:
             f.write(json_str)
             f.write("\n\n")
 
-        return init_img_bk, save_sample_path
+        # return init_img_bk, save_sample_path
+        return save_sample_path
 
 
 controller = AnimateController()
@@ -407,7 +418,7 @@ def ui():
                 ip_adapter_scale,
                 style_dropdown,
             ],
-            outputs=[init_img, result_video]
+            outputs=[result_video]
         )
         default_1.click(
             fn=controller.animate,
@@ -423,7 +434,7 @@ def ui():
                 default_style,
                 example_img,
             ],
-            outputs=[init_img, result_video])
+            outputs=[result_video])
         default_2.click(
             fn=controller.animate,
             inputs=[
@@ -438,7 +449,7 @@ def ui():
                 default_style,
                 example_img,
             ],
-            outputs=[init_img, result_video])
+            outputs=[result_video])
         default_3.click(
             fn=controller.animate,
             inputs=[
@@ -453,7 +464,7 @@ def ui():
                 default_style,
                 example_img,
             ],
-            outputs=[init_img, result_video])
+            outputs=[result_video])
         default_4.click(
             fn=controller.animate,
             inputs=[
@@ -468,7 +479,7 @@ def ui():
                 default_style,
                 example_img,
             ],
-            outputs=[init_img, result_video])
+            outputs=[result_video])
 
         def create_example(input_list):
             return gr.Examples(