rahul7star commited on
Commit
9cd78f6
·
verified ·
1 Parent(s): 617db2b

Update app_g.py

Browse files
Files changed (1) hide show
  1. app_g.py +10 -1
app_g.py CHANGED
@@ -109,12 +109,21 @@ import tempfile
109
 
110
  class PusaVideoDemo:
111
  def __init__(self):
 
 
 
 
 
 
 
 
 
112
  self.device = "cuda" if torch.cuda.is_available() else "cpu"
113
  self.model_manager = None
114
  self.multi_frames_pipe = None
115
  self.v2v_pipe = None
116
  self.t2v_pipe = None
117
- self.base_dir = '/home/user/app/PusaV1/model_zoo/PusaV1/Wan2.1-T2V-14B'
118
 
119
  self.output_dir = "outputs"
120
  os.makedirs(self.output_dir, exist_ok=True)
 
109
 
110
  class PusaVideoDemo:
111
  def __init__(self):
112
+ print("load class demo=======")
113
+ print(WAN_MODEL_PATH)
114
+ print("🔧 Initializing DemoLoader...")
115
+
116
+ # Check WAN model path
117
+ if not os.path.exists(WAN_MODEL_PATH):
118
+ raise FileNotFoundError(f"❌ WAN_MODEL_PATH not found: {WAN_MODEL_PATH}")
119
+
120
+ print(f"✅ WAN_MODEL_PATH resolved: {WAN_MODEL_PATH}")
121
  self.device = "cuda" if torch.cuda.is_available() else "cpu"
122
  self.model_manager = None
123
  self.multi_frames_pipe = None
124
  self.v2v_pipe = None
125
  self.t2v_pipe = None
126
+ self.base_dir = WAN_MODEL_PATH
127
 
128
  self.output_dir = "outputs"
129
  os.makedirs(self.output_dir, exist_ok=True)