bugfix

app.py

@@ -23,9 +23,19 @@ login(token=HF_TOKEN)
 
 bfl_repo = "black-forest-labs/FLUX.1-dev"
 
-pipe = FluxPipeline.from_pretrained("jiuface/lover-snap", use_safetensors=True, torch_dtype=torch.bfloat16)
-# pipe.transformer = transformer
-# pipe.text_encoder_2 = text_encoder_2
+dtype = torch.bfloat16
+
+transformer = FluxTransformer2DModel.from_single_file("https://huggingface.co/jiuface/lover-snap/blob/main/RealFlux_1.0b_Dev_Transformer.safetensors", torch_dtype=dtype)
+quantize(transformer, weights=qfloat8)
+freeze(transformer)
+
+text_encoder_2 = T5EncoderModel.from_pretrained(bfl_repo, subfolder="text_encoder_2", torch_dtype=dtype)
+quantize(text_encoder_2, weights=qfloat8)
+freeze(text_encoder_2)
+
+pipe = FluxPipeline.from_pretrained(bfl_repo, transformer=None, text_encoder_2=None, torch_dtype=dtype)
+pipe.transformer = transformer
+pipe.text_encoder_2 = text_encoder_2
 
 
 pipe.to("cuda")