remove variant

app.py

@@ -82,12 +82,12 @@ def load_pipeline(
     if "ControlNet" in pipeline_to_benchmark:
         controlnet_ckpt = pipeline_details[2]
         controlnet = ControlNetModel.from_pretrained(
-            controlnet_ckpt, variant="fp16", torch_dtype=torch.float16
+            controlnet_ckpt, torch_dtype=torch.float16
         ).to(device)
     elif "Adapters" in pipeline_to_benchmark:
         adapter_clpt = pipeline_details[2]
         adapter = T2IAdapter.from_pretrained(
-            adapter_clpt, variant="fp16", torch_dtype=torch.float16
+            adapter_clpt, torch_dtype=torch.float16
         ).to(device)
 
     # Load pipeline.
@@ -95,9 +95,7 @@ def load_pipeline(
         "ControlNet" not in pipeline_to_benchmark
         or "Adapters" not in pipeline_to_benchmark
     ):
-        pipeline = pipeline_cls.from_pretrained(
-            pipeline_ckpt, variant="fp16", torch_dtype=dtype
-        )
+        pipeline = pipeline_cls.from_pretrained(pipeline_ckpt, torch_dtype=dtype)
 
     elif "ControlNet" in pipeline_to_benchmark:
         pipeline = pipeline_cls.from_pretrained(pipeline_ckpt, controlnet=controlnet)
@@ -205,7 +203,9 @@ def generate(
         print(f"For {num_inference_steps} steps", end_time - start_time)
         print("Avg per step", (end_time - start_time) / num_inference_steps)
 
-    return f"Avg per step: {((end_time - start_time) / num_inference_steps):.4f} seconds."
+    return (
+        f"Avg per step: {((end_time - start_time) / num_inference_steps):.4f} seconds."
+    )
 
 
 with gr.Blocks(css="style.css") as demo: