update inference config
Browse files
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 |
```
|