rahul7star commited on
Commit
566b570
·
verified ·
1 Parent(s): 12a63af

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ import numpy as np
9
 
10
  # Define model path (assuming it's in the HF Space)
11
  MODEL_PATH = "sarthak247/Wan2.1-T2V-1.3B-nf4"
12
- MODEL_FILE = f"{MODEL_PATH}/diffusion_pytorch_model.safetensors"
13
 
14
  # Load model weights manually
15
  device = "cuda" if torch.cuda.is_available() else "cpu"
 
9
 
10
  # Define model path (assuming it's in the HF Space)
11
  MODEL_PATH = "sarthak247/Wan2.1-T2V-1.3B-nf4"
12
+ MODEL_FILE = f"{MODEL_PATH}/blob/main/diffusion_pytorch_model.safetensors"
13
 
14
  # Load model weights manually
15
  device = "cuda" if torch.cuda.is_available() else "cpu"