Update f5_tts/infer/utils_infer.py

f5_tts/infer/utils_infer.py

@@ -138,10 +138,7 @@ asr_pipe = None
 def initialize_asr_pipeline(device: str = device, dtype=None):
     if dtype is None:
         if "cuda" in device:
-            if torch.cuda.get_device_properties(device).major >= 6:
-                dtype = torch.float16
-            else:
-                dtype = torch.float32
+            dtype = torch.float32
         else:
             dtype = torch.float32
     global asr_pipe
@@ -175,10 +172,7 @@ def transcribe(ref_audio, language=None):
 def load_checkpoint(model, ckpt_path, device: str, dtype=None, use_ema=True):
     if dtype is None:
         if "cuda" in device:
-            if torch.cuda.get_device_properties(device).major >= 6:
-                dtype = torch.float16
-            else:
-                dtype = torch.float32
+            dtype = torch.float32
         else:
             dtype = torch.float32
     model = model.to(dtype)