Update app_v5.py

app_v5.py

@@ -60,10 +60,10 @@ transformer_8bit = FluxPipeline.from_pretrained(
 )
 pipe = FluxControlNetPipeline.from_pretrained(
     "LPX55/FLUX.1M-8step_upscaler-cnet",
-    torch_dtype=torch.bfloat16,
-    text_encoder_2=text_encoder_2_8b,
     transformer=transformer_8bit,
-    token=huggingface_token
+    text_encoder_2=text_encoder_2_8b,
+    torch_dtype=torch.float16,
+    device_map="auto",
 )
 pipe.to("cuda")