deepgeek/CosyVoice
2
0
Fork 0
mirror of https://github.com/FunAudioLLM/CosyVoice.git synced 2026-02-05 18:09:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix bug

Browse Source
This commit is contained in:
lyuxiang.lx
2025-04-15 16:36:10 +08:00
parent 37e48dd318
commit 9ebcf7b1ad
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
cosyvoice/bin/inference.py
View File

@@ -63,12 +63,12 @@ def main():
try: try:
with open(args.config, 'r') as f: with open(args.config, 'r') as f:
configs = load_hyperpyyaml(f, overrides={'qwen_pretrain_path': args.qwen_pretrain_path}) configs = load_hyperpyyaml(f, overrides={'qwen_pretrain_path': args.qwen_pretrain_path})
model = CosyVoice2Model(configs['llm'], configs['flow'], configs['hift'], fp16=False) model = CosyVoice2Model(configs['llm'], configs['flow'], configs['hift'])
except Exception: except Exception:
try: try:
with open(args.config, 'r') as f: with open(args.config, 'r') as f:
configs = load_hyperpyyaml(f) configs = load_hyperpyyaml(f)
model = CosyVoiceModel(configs['llm'], configs['flow'], configs['hift'], fp16=False) model = CosyVoiceModel(configs['llm'], configs['flow'], configs['hift'])
except Exception: except Exception:
raise TypeError('no valid model_type!') raise TypeError('no valid model_type!')

6
cosyvoice/cli/model.py
View File

@@ -30,7 +30,7 @@ class CosyVoiceModel:
llm: torch.nn.Module, llm: torch.nn.Module,
flow: torch.nn.Module, flow: torch.nn.Module,
hift: torch.nn.Module, hift: torch.nn.Module,
fp16: bool): fp16: bool = False):
self.device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') self.device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
self.llm = llm self.llm = llm
self.flow = flow self.flow = flow
@@ -240,8 +240,8 @@ class CosyVoice2Model(CosyVoiceModel):
llm: torch.nn.Module, llm: torch.nn.Module,
flow: torch.nn.Module, flow: torch.nn.Module,
hift: torch.nn.Module, hift: torch.nn.Module,
fp16: bool, fp16: bool = False,
use_flow_cache: bool): use_flow_cache: bool = False):
self.device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') self.device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
self.llm = llm self.llm = llm
self.flow = flow self.flow = flow