jieliu commited on
Commit
d4977a2
·
verified ·
1 Parent(s): 5efdfea

update inference config

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -35,6 +35,6 @@ pipe.transformer = pipe.transformer.merge_and_unload()
35
  pipe = pipe.to(device)
36
 
37
  prompt = 'a photo of a black kite and a green bear'
38
- image = pipe(prompt, height=512, width=512, num_inference_steps=40,guidance_scale=4.5).images[0]
39
  image.save(f"flow_grpo.png")
40
  ```
 
35
  pipe = pipe.to(device)
36
 
37
  prompt = 'a photo of a black kite and a green bear'
38
+ image = pipe(prompt, height=512, width=512, num_inference_steps=40,guidance_scale=4.5,negative_prompt="").images[0]
39
  image.save(f"flow_grpo.png")
40
  ```