Update app.py

app.py

@@ -40,13 +40,12 @@ text_encoder_2_4bit = T5EncoderModel.from_pretrained(
 #     bnb_4bit_use_double_quant=True,
 # )
 
-# transformer_4bit = FluxTransformer2DModel.from_pretrained(
-#     "black-forest-labs/FLUX.1-dev",
-#     subfolder="transformer",
-#     quantization_config=quant_config,
-#     torch_dtype=torch.bfloat16,
-#     token=HF_TOKEN
-# )
+transformerx = FluxTransformer2DModel.from_pretrained(
+    "black-forest-labs/FLUX.1-dev",
+    subfolder="transformer",
+    torch_dtype=torch.bfloat16,
+    token=HF_TOKEN
+)
 # text_encoder_8bit = T5EncoderModel.from_pretrained(
 #     "black-forest-labs/FLUX.1-dev",
 #     subfolder="text_encoder_2",
@@ -67,7 +66,7 @@ pipe = FluxControlNetInpaintingPipeline.from_pretrained(
     "black-forest-labs/FLUX.1-dev",
     controlnet=controlnet,
     # text_encoder_2=text_encoder_8bit,
-    # transformer=transformer_4bit,
+    transformer=transformerx,
     torch_dtype=torch.bfloat16, 
     # device_map="balanced",