Spaces:
Running
Running
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.
|
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
|
|