Update app.py

app.py

@@ -39,15 +39,15 @@ good_vae = AutoencoderKL.from_pretrained("black-forest-labs/FLUX.1-dev",
 #     token=HF_TOKEN
 # )
 
-# # Quantize the text encoder to 8-bit precision
-# quant_config = BitsAndBytesConfig(load_in_8bit=True)
-# text_encoder_8bit = T5EncoderModel.from_pretrained(
-#     "black-forest-labs/FLUX.1-dev",
-#     subfolder="text_encoder_2",
-#     quantization_config=quant_config,
-#     torch_dtype=torch.float16,
-#     token=HF_TOKEN
-# )
+# Quantize the text encoder to 8-bit precision
+quant_config = BitsAndBytesConfig(load_in_8bit=True)
+text_encoder_8bit = T5EncoderModel.from_pretrained(
+    "black-forest-labs/FLUX.1-dev",
+    subfolder="text_encoder_2",
+    quantization_config=quant_config,
+    torch_dtype=torch.float16,
+    token=HF_TOKEN
+)
 
 # # Load necessary models and processors
 # controlnet = FluxControlNetModel.from_pretrained("alimama-creative/FLUX.1-dev-Controlnet-Inpainting-Beta", torch_dtype=torch.bfloat16)