Fix lora path

app.py

@@ -19,14 +19,18 @@ if LOCAL:
     lora_file_path = '/share2/duanyuxuan/diff_playground/SVD-TDD/svd-xt-1-1_tdd_lora_weights.safetensors'
 else:
     svd_path = 'stabilityai/stable-video-diffusion-img2vid-xt-1-1'
-    lora_file_path = 'RED-AIGC/TDD/svd-xt-1-1_tdd_lora_weights.safetensors'
+    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_file_path)
+    if LOCAL:
+        load_lora_weights(pipeline.unet, lora_file_path)
+    else:
+        load_lora_weights(pipeline.unet, lora_repo_path, weight_name = lora_weight_name)
 
 max_64_bit_int = 2**63 - 1