fix: device placements.

app.py

@@ -82,13 +82,11 @@ def load_pipeline(
     if "ControlNet" in pipeline_to_benchmark:
         controlnet_ckpt = pipeline_details[2]
         controlnet = ControlNetModel.from_pretrained(
-            controlnet_ckpt, torch_dtype=torch.float16
+            controlnet_ckpt, torch_dtype=dtype
         ).to(device)
     elif "Adapters" in pipeline_to_benchmark:
         adapter_clpt = pipeline_details[2]
-        adapter = T2IAdapter.from_pretrained(
-            adapter_clpt, torch_dtype=torch.float16
-        ).to(device)
+        adapter = T2IAdapter.from_pretrained(adapter_clpt, torch_dtype=dtype).to(device)
 
     # Load pipeline.
     if (
@@ -98,9 +96,13 @@ def load_pipeline(
         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)
+        pipeline = pipeline_cls.from_pretrained(
+            pipeline_ckpt, controlnet=controlnet, torch_dtype=dtype
+        )
     elif "Adapters" in pipeline_to_benchmark:
-        pipeline = pipeline_cls.from_pretrained(pipeline_ckpt, adapter=adapter)
+        pipeline = pipeline_cls.from_pretrained(
+            pipeline_ckpt, adapter=adapter, torch_dtype=dtype
+        )
 
     pipeline.to(device)