lint(api): consistently use server dtype

ssube · Apr 22, 2023 · 9e9feb2 · 9e9feb2
1 parent 9bb612f
commit 9e9feb2
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/api/onnx_web/diffusers/load.py b/api/onnx_web/diffusers/load.py
@@ -133,9 +133,7 @@ def load_pipeline(
     loras = loras or []
     control_key = control.name if control is not None else None
 
-    torch_dtype = (
-        torch.float16 if "torch-fp16" in server.optimizations else torch.float32
-    )
+    torch_dtype = server.torch_dtype()
     logger.debug("using Torch dtype %s for pipeline", torch_dtype)
     pipe_key = (
         pipeline,