Update README.md

README.md

@@ -79,7 +79,7 @@ for i in range(len((dev_shards))):
 merged_state_dict.update(guidance_state_dict)
 load_model_dict_into_meta(model, merged_state_dict)
 
-model.to(torch.bfloat16).save_pretrained("/raid/.cache/huggingface/merged-flux")
+model.to(torch.bfloat16).save_pretrained("merged-flux")
 ```
 
 ## Inference code
@@ -88,14 +88,13 @@ model.to(torch.bfloat16).save_pretrained("/raid/.cache/huggingface/merged-flux")
 from diffusers import FluxPipeline, FluxTransformer2DModel
 import torch
 
-transformer = FluxTransformer2DModel.from_pretrained("sayakpaul/FLUX.1-merged", torch_dtype=torch.bfloat16)
 pipeline = FluxPipeline.from_pretrained(
-    "black-forest-labs/FLUX.1-dev", transformer=transformer, torch_dtype=torch.bfloat16
+    "sayakpaul/FLUX.1-merged", transformer=transformer, torch_dtype=torch.bfloat16
 ).to("cuda")
 image = pipeline(
     prompt="a tiny astronaut hatching from an egg on the moon",
     guidance_scale=3.5,
-    num_inference_steps=50,
+    num_inference_steps=4,
     height=880,
     width=1184,
     max_sequence_length=512,