.

options/Video_model/Model.py

@@ -12,11 +12,11 @@ svd_path = 'stabilityai/stable-video-diffusion-img2vid-xt-1-1'
 lora_repo_path = 'RED-AIGC/TDD'
 lora_weight_name = 'svd-xt-1-1_tdd_lora_weights.safetensors'
 
-if torch.cuda.is_available():
-    noise_scheduler = TDDSVDStochasticIterativeScheduler(num_train_timesteps = 250, sigma_min = 0.002, sigma_max = 700.0, sigma_data = 1.0, 
-                                                        s_noise = 1.0, rho = 7, clip_denoised = False)
-    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)
+# if torch.cuda.is_available():
+noise_scheduler = TDDSVDStochasticIterativeScheduler(num_train_timesteps = 250, sigma_min = 0.002, sigma_max = 700.0, sigma_data = 1.0, 
+                                                    s_noise = 1.0, rho = 7, clip_denoised = False)
+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(