Undo float16/half precision for cpu

Darwinkel · Apr 2, 2024 · 63ef2aa · 63ef2aa
1 parent 1febf56
commit 63ef2aa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/shepardtts/utils.py b/shepardtts/utils.py
@@ -33,7 +33,7 @@ def load_checkpoint() -> ShepardXtts:
     if settings.DEVICE == "cpu":
         import intel_extension_for_pytorch as ipex
 
-        model = ipex.optimize(model, weights_prepack=False, dtype=torch.half)
+        model = ipex.optimize(model, weights_prepack=False)
         model = torch.compile(model, backend="ipex")
 
     return model