OmPrakashSingh1704 commited on
Commit
b8becf0
·
1 Parent(s): 33e0753
options/Video_model/Model.py CHANGED
@@ -19,7 +19,7 @@ noise_scheduler = TDDSVDStochasticIterativeScheduler(
19
  sigma_data=1.0, s_noise=1.0, rho=7, clip_denoised=False
20
  )
21
  pipeline = StableVideoDiffusionPipeline.from_pretrained(
22
- svd_path, scheduler=noise_scheduler, torch_dtype=torch.float16, variant="fp16"
23
  ).to(device)
24
  load_lora_weights(pipeline.unet, lora_repo_path, weight_name=lora_weight_name)
25
 
 
19
  sigma_data=1.0, s_noise=1.0, rho=7, clip_denoised=False
20
  )
21
  pipeline = StableVideoDiffusionPipeline.from_pretrained(
22
+ svd_path, scheduler=noise_scheduler, torch_dtype=torch.float32
23
  ).to(device)
24
  load_lora_weights(pipeline.unet, lora_repo_path, weight_name=lora_weight_name)
25
 
options/Video_model/__pycache__/Model.cpython-310.pyc CHANGED
Binary files a/options/Video_model/__pycache__/Model.cpython-310.pyc and b/options/Video_model/__pycache__/Model.cpython-310.pyc differ