Zhouyan248 commited on
Commit
231fb10
1 Parent(s): a876b6c

Update base/text_to_video/__init__.py

Browse files
Files changed (1) hide show
  1. base/text_to_video/__init__.py +2 -2
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
- # torch.manual_seed(seed)
43
- # torch.cuda.manual_seed_all(seed)
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