Spaces:

OmPrakashSingh1704
/

ADVERTISE

Running

OmPrakashSingh1704 commited on Sep 17, 2024

Commit

1b418ed

1 Parent(s): 0056a56

.

Files changed (2) hide show

options/Video_model/Model.py CHANGED Viewed

@@ -1,4 +1,4 @@
-import spaces
 import torch
 from diffusers import StableVideoDiffusionPipeline
 from diffusers.utils import load_image
@@ -16,7 +16,7 @@ if torch.cuda.is_available():
     pipeline = StableVideoDiffusionPipeline.from_pretrained(svd_path, scheduler = noise_scheduler, torch_dtype = torch.float16, variant = "fp16").to('cuda')
     load_lora_weights(pipeline.unet, lora_repo_path, weight_name = lora_weight_name)
-@spaces.GPU
 def Video(
     image: Image,
     seed: Optional[int] = 1,

+# import spaces
 import torch
 from diffusers import StableVideoDiffusionPipeline
 from diffusers.utils import load_image
     pipeline = StableVideoDiffusionPipeline.from_pretrained(svd_path, scheduler = noise_scheduler, torch_dtype = torch.float16, variant = "fp16").to('cuda')
     load_lora_weights(pipeline.unet, lora_repo_path, weight_name = lora_weight_name)
+# @spaces.GPU
 def Video(
     image: Image,
     seed: Optional[int] = 1,

options/Video_model/__pycache__/Model.cpython-310.pyc CHANGED Viewed

Binary files a/options/Video_model/__pycache__/Model.cpython-310.pyc and b/options/Video_model/__pycache__/Model.cpython-310.pyc differ