Merge pull request #246 from sharziki/fix/unclosed-file-handles
fix: close file handles in from_local() config loading
This commit is contained in:
@@ -852,7 +852,8 @@ class VoxCPMModel(nn.Module):
|
||||
device: str | None = 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)
|
||||
audio_vae_config = getattr(config, "audio_vae_config", None)
|
||||
audio_vae = AudioVAE(config=audio_vae_config) if audio_vae_config else AudioVAE()
|
||||
|
||||
@@ -1103,7 +1103,8 @@ class VoxCPM2Model(nn.Module):
|
||||
device: str | None = 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)
|
||||
audio_vae_config = getattr(config, "audio_vae_config", None)
|
||||
audio_vae = AudioVAEV2(config=audio_vae_config) if audio_vae_config else AudioVAEV2()
|
||||
|
||||
Reference in New Issue
Block a user