fix: sanitize name field and remove double load_npz call
- _resolve_named_path: replace / \ and null in name to prevent path traversal outside cache_dir (would cause a confusing FileNotFoundError at np.savez time instead of at path resolution). - train_lora: load_npz was called twice per clip when prompt was in prompts.txt; consolidate to a single call before prompt resolution. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
+2
-6
@@ -264,13 +264,9 @@ def main():
|
||||
print(f" [LoRA] Warning: no audio file found for {npz_path.name} — skipping")
|
||||
continue
|
||||
|
||||
bundle = load_npz(npz_path)
|
||||
# Prompt priority: prompts.txt override > embedded in .npz > directory name
|
||||
prompt = prompt_map.get(npz_path.name)
|
||||
if prompt is None:
|
||||
bundle = load_npz(npz_path)
|
||||
prompt = bundle.get("prompt", default_prompt)
|
||||
else:
|
||||
bundle = load_npz(npz_path)
|
||||
prompt = prompt_map.get(npz_path.name, bundle.get("prompt", default_prompt))
|
||||
|
||||
print(f" {npz_path.name} + {audio_path.name}: '{prompt}'")
|
||||
|
||||
|
||||
Reference in New Issue
Block a user