Spaces:
Paused
Paused
Update app_g.py
Browse files
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 =
|
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)
|