Update README.md

README.md

@@ -106,12 +106,12 @@ class FluxGenerator:
 
 T5EncoderModel.from_config = lambda c: T5EncoderModel(c).to(dtype=torch.bfloat16)
 t5 = T5Model.from_pretrained('./flux-t5')._wrapped
-transformer = FluxTransformer2DModel.from_pretrained('flux-fp8-e4m3fn',
+transformer = FluxTransformer2DModel.from_pretrained('twodgirl/flux-dev-fp8-e4m3fn-diffusers',
                                                      subfolder='transformer',
                                                      torch_dtype=torch.float8_e4m3fn)
 replace_regular_linears(transformer)
-vae = AutoencoderKL.from_pretrained('flux', subfolder='vae').to(torch.bfloat16)
-pipe = FluxPipeline.from_pretrained('flux',
+vae = AutoencoderKL.from_pretrained('black-forest-labs/FLUX.1-dev', subfolder='vae').to(torch.bfloat16)
+pipe = FluxPipeline.from_pretrained('black-forest-labs/FLUX.1-dev',
                                     text_encoder_2=t5,
                                     transformer=transformer,
                                     vae=vae)