multi-gpu need fixed seed

Author: Howe2018

generate_video.py

@@ -41,7 +41,7 @@
     parser.add_argument("--use_usp", action="store_true")
     parser.add_argument("--offload", action="store_true")
     parser.add_argument("--fps", type=int, default=24)
-    parser.add_argument("--seed", type=int, default=-1)
+    parser.add_argument("--seed", type=int, default=None)
     parser.add_argument(
         "--prompt",
         type=str,
@@ -54,7 +54,7 @@
     print("model_id:", args.model_id)
 
     assert (args.use_usp and args.seed is not None) or (not args.use_usp), "usp mode need seed"
-    if args.seed == -1:
+    if args.seed is None:
         random.seed(time.time())
         args.seed = int(random.randrange(4294967294))
 

generate_video_df.py

@@ -32,7 +32,7 @@
     parser.add_argument("--use_usp", action="store_true")
     parser.add_argument("--offload", action="store_true")
     parser.add_argument("--fps", type=int, default=24)
-    parser.add_argument("--seed", type=int, default=-1)
+    parser.add_argument("--seed", type=int, default=None)
     parser.add_argument(
         "--prompt",
         type=str,
@@ -45,7 +45,7 @@
     print("model_id:", args.model_id)
 
     assert (args.use_usp and args.seed is not None) or (not args.use_usp), "usp mode need seed"
-    if args.seed == -1:
+    if args.seed is None:
         random.seed(time.time())
         args.seed = int(random.randrange(4294967294))