AnyaSchen commited on
Commit
30744bc
·
1 Parent(s): d3fe52c

fix: print using model 2

Browse files
whisper_streaming_custom/backends.py CHANGED
@@ -143,7 +143,7 @@ class FasterWhisperASR(ASRBase):
143
  device = "cuda" if torch and torch.cuda.is_available() else "cpu"
144
  compute_type = "float16" if device == "cuda" else "float32"
145
 
146
- logger.info(f"Loading whisper model {model_size_or_path} on {device} with compute type {compute_type}")
147
 
148
  model = WhisperModel(
149
  model_size_or_path,
 
143
  device = "cuda" if torch and torch.cuda.is_available() else "cpu"
144
  compute_type = "float16" if device == "cuda" else "float32"
145
 
146
+ print(f"Loading whisper model {model_size_or_path} on {device} with compute type {compute_type}")
147
 
148
  model = WhisperModel(
149
  model_size_or_path,