bugfix

app.py

@@ -22,19 +22,10 @@ login(token=HF_TOKEN)
 
 
 bfl_repo = "black-forest-labs/FLUX.1-dev"
-dtype = torch.bfloat16
-transformer_url = "https://huggingface.co/jiuface/lover-snap/RealFlux_1.0b_Dev_Transformer.safetensors"
-transformer = FluxTransformer2DModel.from_single_file(transformer_url, 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 = FluxPipeline.from_pretrained("jiuface/lover-snap/fluxRealistic.safetensors", torch_dtype=dtype)
+# pipe.transformer = transformer
+# pipe.text_encoder_2 = text_encoder_2
 
 
 pipe.to("cuda")