Fix: don't use rank-768 LoRA alongside full distil model; capture stderr
Browse files
app.py
CHANGED
|
@@ -25,8 +25,7 @@ FINETUNES_DIR = WAN2GP_ROOT / "finetunes"
|
|
| 25 |
GENERATE_PY = Path(__file__).parent / "generate.py"
|
| 26 |
|
| 27 |
SULPHUR_ASSETS = [
|
| 28 |
-
("SulphurAI/Sulphur-2-base", "sulphur_distil_bf16.safetensors",
|
| 29 |
-
("SulphurAI/Sulphur-2-base", "sulphur_lora_rank_768.safetensors", LORAS_DIR),
|
| 30 |
]
|
| 31 |
LTX_ASSETS = [
|
| 32 |
("DeepBeepMeep/LTX-2", "ltx-2.3-22b-distilled-lora-384-1.1.safetensors", LORAS_DIR),
|
|
@@ -42,8 +41,9 @@ SULPHUR_FINETUNE = {
|
|
| 42 |
"architecture": "ltx2_22B",
|
| 43 |
"parent_model_type": "ltx2_22B",
|
| 44 |
"description": "LTX-2.3 fine-tuned i2v. Distilled checkpoint.",
|
|
|
|
| 45 |
"URLs": [str(CKPTS_DIR / "sulphur_distil_bf16.safetensors")],
|
| 46 |
-
"preload_URLs": [
|
| 47 |
},
|
| 48 |
"num_inference_steps": 8,
|
| 49 |
"video_length": 81,
|
|
@@ -125,12 +125,17 @@ def _generate(image: str, prompt: str, model: str, steps: int,
|
|
| 125 |
]
|
| 126 |
|
| 127 |
log_lines = []
|
| 128 |
-
proc = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.
|
| 129 |
text=True, bufsize=1, env=env)
|
| 130 |
for line in proc.stdout:
|
| 131 |
log_lines.append(line.rstrip())
|
| 132 |
print(line, end="")
|
|
|
|
| 133 |
proc.wait()
|
|
|
|
|
|
|
|
|
|
|
|
|
| 134 |
|
| 135 |
log = "\n".join(log_lines)
|
| 136 |
if proc.returncode != 0 or not out_file.exists():
|
|
|
|
| 25 |
GENERATE_PY = Path(__file__).parent / "generate.py"
|
| 26 |
|
| 27 |
SULPHUR_ASSETS = [
|
| 28 |
+
("SulphurAI/Sulphur-2-base", "sulphur_distil_bf16.safetensors", CKPTS_DIR),
|
|
|
|
| 29 |
]
|
| 30 |
LTX_ASSETS = [
|
| 31 |
("DeepBeepMeep/LTX-2", "ltx-2.3-22b-distilled-lora-384-1.1.safetensors", LORAS_DIR),
|
|
|
|
| 41 |
"architecture": "ltx2_22B",
|
| 42 |
"parent_model_type": "ltx2_22B",
|
| 43 |
"description": "LTX-2.3 fine-tuned i2v. Distilled checkpoint.",
|
| 44 |
+
# Full distilled model — do NOT also preload the rank-768 LoRA (README: use one or the other)
|
| 45 |
"URLs": [str(CKPTS_DIR / "sulphur_distil_bf16.safetensors")],
|
| 46 |
+
"preload_URLs": [],
|
| 47 |
},
|
| 48 |
"num_inference_steps": 8,
|
| 49 |
"video_length": 81,
|
|
|
|
| 125 |
]
|
| 126 |
|
| 127 |
log_lines = []
|
| 128 |
+
proc = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE,
|
| 129 |
text=True, bufsize=1, env=env)
|
| 130 |
for line in proc.stdout:
|
| 131 |
log_lines.append(line.rstrip())
|
| 132 |
print(line, end="")
|
| 133 |
+
stderr_out = proc.stderr.read()
|
| 134 |
proc.wait()
|
| 135 |
+
if stderr_out:
|
| 136 |
+
log_lines.append("--- stderr ---")
|
| 137 |
+
log_lines.extend(stderr_out.splitlines())
|
| 138 |
+
print(stderr_out)
|
| 139 |
|
| 140 |
log = "\n".join(log_lines)
|
| 141 |
if proc.returncode != 0 or not out_file.exists():
|