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)
|