Daankular commited on
Commit
4fe229d
·
1 Parent(s): 031028a

Fix: don't use rank-768 LoRA alongside full distil model; capture stderr

Browse files
Files changed (1) hide show
  1. app.py +9 -4
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", CKPTS_DIR),
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": [str(LORAS_DIR / "sulphur_lora_rank_768.safetensors") + "|%lora_dir"],
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.STDOUT,
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():