Update app.py

app.py

@@ -22,6 +22,10 @@ from optimum.quanto import freeze, qfloat8, quantize
 #quantize(controlnet, weights=qfloat8)
 #freeze(controlnet)
 
+transformer = FluxTransformer2DModel.from_pretrained(
+    "black-forest-labs/FLUX.1-dev", subfolder='transformer', torch_dtype=torch.bfloat16
+)
+
 text_encoder_2 = T5EncoderModel.from_pretrained("black-forest-labs/FLUX.1-dev", subfolder="text_encoder_2", torch_dtype=torch.bfloat16)
 quantize(text_encoder_2, weights=qfloat8)
 freeze(text_encoder_2)
@@ -29,6 +33,7 @@ freeze(text_encoder_2)
 pipe = FluxControlNetInpaintingPipeline.from_pretrained(
     "black-forest-labs/FLUX.1-dev",
     text_encoder_2=None,
+    transformer=transformer,
     torch_dtype=torch.bfloat16
 )
 pipe.text_encoder_2 = text_encoder_2