Spaces:
Running
on
A10G
Running
on
A10G
Zhouyan248
commited on
Commit
•
231fb10
1
Parent(s):
a876b6c
Update base/text_to_video/__init__.py
Browse files
base/text_to_video/__init__.py
CHANGED
@@ -39,8 +39,8 @@ def model_t2v_fun(args):
|
|
39 |
return VideoGenPipeline(vae=vae, text_encoder=text_encoder_one, tokenizer=tokenizer_one, scheduler=scheduler, unet=unet)
|
40 |
|
41 |
def setup_seed(seed):
|
42 |
-
|
43 |
-
|
44 |
random.seed(seed)
|
45 |
|
46 |
|
|
|
39 |
return VideoGenPipeline(vae=vae, text_encoder=text_encoder_one, tokenizer=tokenizer_one, scheduler=scheduler, unet=unet)
|
40 |
|
41 |
def setup_seed(seed):
|
42 |
+
torch.manual_seed(seed)
|
43 |
+
torch.cuda.manual_seed_all(seed)
|
44 |
random.seed(seed)
|
45 |
|
46 |
|