Update f5_tts/infer/utils_infer.py

f5_tts/infer/utils_infer.py

@@ -138,9 +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.is_bf16_supported():
-                dtype = torch.bfloat16
-            elif torch.cuda.get_device_properties(device).major >= 6:
+            if torch.cuda.get_device_properties(device).major >= 6:
                 dtype = torch.float16
             else:
                 dtype = torch.float32
@@ -177,9 +175,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.is_bf16_supported():
-                dtype = torch.bfloat16
-            elif torch.cuda.get_device_properties(device).major >= 6:
+            if torch.cuda.get_device_properties(device).major >= 6:
                 dtype = torch.float16
             else:
                 dtype = torch.float32