FireRedTeam
/

FireRed-Image-Edit-1.0-Lightning

FireRed-Image-Edit-1.0

Model card Files Files and versions

FireRedTeam commited on Mar 5

Commit

454b1b1

·

verified ·

1 Parent(s): 94a9271

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -23,7 +23,7 @@ from diffusers import QwenImageEditPlusPipeline
 import torch
 from PIL import Image
-pipe = pipe = QwenImageEditPlusPipeline.from_pretrained(
     "FireRedTeam/FireRed-Image-Edit-1.0", torch_dtype=torch.bfloat16,
 ).to("cuda")
 pipe.load_lora_weights(
@@ -36,7 +36,7 @@ input_image_raw = Image.open(input_image_path).convert('RGB')
 image = pipe(
     image = [input_image_raw],
-    prompt = instruction,
     height = None,
     width = None,
     num_inference_steps = 8,

 import torch
 from PIL import Image
+pipe = QwenImageEditPlusPipeline.from_pretrained(
     "FireRedTeam/FireRed-Image-Edit-1.0", torch_dtype=torch.bfloat16,
 ).to("cuda")
 pipe.load_lora_weights(
 image = pipe(
     image = [input_image_raw],
+    prompt = prompt,
     height = None,
     width = None,
     num_inference_steps = 8,