Pine/VoxCPM
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #246 from sharziki/fix/unclosed-file-handles

Browse Source 
fix: close file handles in from_local() config loading
This commit is contained in:
xliucs
2026-04-11 13:10:04 +08:00
committed by GitHub
parent e4e049624c fb46aad9a5
commit 5510503182
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/voxcpm/model/voxcpm.py
+2 -1
View File
@@ -852,7 +852,8 @@ class VoxCPMModel(nn.Module):
device: str | None = None, device: str | None = None,
lora_config: LoRAConfig = None, lora_config: LoRAConfig = None,
): ):
config = VoxCPMConfig.model_validate_json(open(os.path.join(path, "config.json")).read()) with open(os.path.join(path, "config.json"), "r", encoding="utf-8") as _cfg_f:
config = VoxCPMConfig.model_validate_json(_cfg_f.read())
tokenizer = LlamaTokenizerFast.from_pretrained(path) tokenizer = LlamaTokenizerFast.from_pretrained(path)
audio_vae_config = getattr(config, "audio_vae_config", None) audio_vae_config = getattr(config, "audio_vae_config", None)
audio_vae = AudioVAE(config=audio_vae_config) if audio_vae_config else AudioVAE() audio_vae = AudioVAE(config=audio_vae_config) if audio_vae_config else AudioVAE()
src/voxcpm/model/voxcpm2.py
+2 -1
View File
@@ -1103,7 +1103,8 @@ class VoxCPM2Model(nn.Module):
device: str | None = None, device: str | None = None,
lora_config: LoRAConfig = None, lora_config: LoRAConfig = None,
): ):
config = VoxCPMConfig.model_validate_json(open(os.path.join(path, "config.json")).read()) with open(os.path.join(path, "config.json"), "r", encoding="utf-8") as _cfg_f:
config = VoxCPMConfig.model_validate_json(_cfg_f.read())
tokenizer = LlamaTokenizerFast.from_pretrained(path) tokenizer = LlamaTokenizerFast.from_pretrained(path)
audio_vae_config = getattr(config, "audio_vae_config", None) audio_vae_config = getattr(config, "audio_vae_config", None)
audio_vae = AudioVAEV2(config=audio_vae_config) if audio_vae_config else AudioVAEV2() audio_vae = AudioVAEV2(config=audio_vae_config) if audio_vae_config else AudioVAEV2()