FireRedTeam commited on
Commit
454b1b1
·
verified ·
1 Parent(s): 94a9271

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -23,7 +23,7 @@ from diffusers import QwenImageEditPlusPipeline
23
  import torch
24
  from PIL import Image
25
 
26
- pipe = pipe = QwenImageEditPlusPipeline.from_pretrained(
27
  "FireRedTeam/FireRed-Image-Edit-1.0", torch_dtype=torch.bfloat16,
28
  ).to("cuda")
29
  pipe.load_lora_weights(
@@ -36,7 +36,7 @@ input_image_raw = Image.open(input_image_path).convert('RGB')
36
 
37
  image = pipe(
38
  image = [input_image_raw],
39
- prompt = instruction,
40
  height = None,
41
  width = None,
42
  num_inference_steps = 8,
 
23
  import torch
24
  from PIL import Image
25
 
26
+ pipe = QwenImageEditPlusPipeline.from_pretrained(
27
  "FireRedTeam/FireRed-Image-Edit-1.0", torch_dtype=torch.bfloat16,
28
  ).to("cuda")
29
  pipe.load_lora_weights(
 
36
 
37
  image = pipe(
38
  image = [input_image_raw],
39
+ prompt = prompt,
40
  height = None,
41
  width = None,
42
  num_inference_steps = 8,