fanhuadesenlinnn/VoxCPM
1
0
Fork 0
mirror of https://github.com/OpenBMB/VoxCPM.git synced 2026-09-20 10:53:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #329 from zhuxiaoxuhit/fix/lora-config-autoload

Browse Source 
Fix load saved lora_config.json when loading LoRA weights
This commit is contained in:
ZGY
2026-06-29 13:14:01 +08:00
committed by GitHub
parent 2b26f1ff8d ae8fa7e0b4
commit e1e9a6a788
1 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
src/voxcpm/core.py
View File

@@ -45,14 +45,17 @@ class VoxCPM:
file=sys.stderr,
)
# If lora_weights_path is provided but no lora_config, create a default one
# If lora_weights_path is provided but no lora_config, load the saved
# lora_config.json (so r/alpha match the checkpoint); else use a default.
if lora_weights_path is not None and lora_config is None:
lora_config = LoRAConfig(
enable_lm=True,
enable_dit=True,
enable_proj=False,
)
print(f"Auto-created default LoRAConfig for loading weights from: {lora_weights_path}", file=sys.stderr)
cfg_path = os.path.join(lora_weights_path, "lora_config.json")
if os.path.isdir(lora_weights_path) and os.path.isfile(cfg_path):
with open(cfg_path, "r", encoding="utf-8") as f:
lora_config = LoRAConfig(**json.load(f)["lora_config"])
print(f"Loaded LoRAConfig from: {cfg_path}", file=sys.stderr)
else:
lora_config = LoRAConfig(enable_lm=True, enable_dit=True, enable_proj=False)
print(f"Auto-created default LoRAConfig for loading weights from: {lora_weights_path}", file=sys.stderr)
# Determine model type from config.json architecture field
config_path = os.path.join(voxcpm_model_path, "config.json")