sayakpaul
/

FLUX.1-merged

Inference Endpoints

Model card Files Files and versions Community

sayakpaul HF staff commited on Aug 10

Commit

f37b438

•

1 Parent(s): 7b3d315

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -85,11 +85,11 @@ model.to(torch.bfloat16).save_pretrained("merged-flux")
 ## Inference code
 ```python
-from diffusers import FluxPipeline, FluxTransformer2DModel
 import torch
 pipeline = FluxPipeline.from_pretrained(
-    "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",

 ## Inference code
 ```python
+from diffusers import FluxPipeline
 import torch
 pipeline = FluxPipeline.from_pretrained(
+    "sayakpaul/FLUX.1-merged", torch_dtype=torch.bfloat16
 ).to("cuda")
 image = pipeline(
     prompt="a tiny astronaut hatching from an egg on the moon",