Update README.md

README.md

@@ -30,7 +30,7 @@ def load_pipeline(four_bit=False):
             "sayakpaul/FLUX.1-Fill-dev-nf4", subfolder="text_encoder_2", torch_dtype=torch.bfloat16
         )
         pipeline = FluxFillPipeline.from_pipe(
-            orig_pipeline, ransformer=transformer, text_encoder_2=text_encoder_2, torch_dtype=torch.bfloat16
+            orig_pipeline, transformer=transformer, text_encoder_2=text_encoder_2, torch_dtype=torch.bfloat16
         )
     else:
         transformer = FluxTransformer2DModel.from_pretrained(
@@ -51,8 +51,9 @@ def load_conditions():
     return image, mask
 
 
-def main(ckpt_id: str, four_bit: bool = False):
-    pipe = load_pipeline(ckpt_id=ckpt_id, four_bit=four_bit)
+def main(four_bit: bool = False):
+    pipe = load_pipeline(four_bit=four_bit)
+    ckpt_id = "sayakpaul/FLUX.1-Fill-dev-nf4"
     image, mask = load_conditions()
     image = pipe(
         prompt="a white paper cup",