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

fix: use specific exceptions instead of bare except

Browse Source 
- lora_ft_webui.py: except (JSONDecodeError, OSError) for config file
- voxcpm.py: except ImportError for triton availability check
This commit is contained in:
haosenwang1018
2026-02-24 22:19:45 +00:00
parent acaadb19e9
commit 8df79de636
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
lora_ft_webui.py
+1 -1
View File
@@ -156,7 +156,7 @@ def scan_lora_checkpoints(root_dir="lora", with_info=False):
with open(lora_config_file, "r", encoding="utf-8") as f: with open(lora_config_file, "r", encoding="utf-8") as f:
lora_info = json.load(f) lora_info = json.load(f)
base_model = lora_info.get("base_model", "Unknown") base_model = lora_info.get("base_model", "Unknown")
except: except (json.JSONDecodeError, OSError):
pass pass
checkpoints.append((rel_path, base_model)) checkpoints.append((rel_path, base_model))
else: else:
src/voxcpm/model/voxcpm.py
+1 -1
View File
@@ -222,7 +222,7 @@ class VoxCPMModel(nn.Module):
raise ValueError("VoxCPMModel can only be optimized on CUDA device") raise ValueError("VoxCPMModel can only be optimized on CUDA device")
try: try:
import triton import triton
except: except ImportError:
raise ValueError("triton is not installed") raise ValueError("triton is not installed")
self.base_lm.forward_step = torch.compile(self.base_lm.forward_step, mode="reduce-overhead", fullgraph=True) self.base_lm.forward_step = torch.compile(self.base_lm.forward_step, mode="reduce-overhead", fullgraph=True)
self.residual_lm.forward_step = torch.compile(self.residual_lm.forward_step, mode="reduce-overhead", fullgraph=True) self.residual_lm.forward_step = torch.compile(self.residual_lm.forward_step, mode="reduce-overhead", fullgraph=True)