Update README.md
Browse files
README.md
CHANGED
|
@@ -33,6 +33,7 @@ prompt = "mtg style Tony Stark, in a city"
|
|
| 33 |
with torch.cuda.amp.autocast(True):
|
| 34 |
image = pipeline(prompt, num_inference_steps=40, guidance_scale=8, height = 512, width = 712, scheduler = "DDIM").images[0]
|
| 35 |
image.save(f"img.png")
|
|
|
|
| 36 |
```
|
| 37 |
|
| 38 |
|
|
|
|
| 33 |
with torch.cuda.amp.autocast(True):
|
| 34 |
image = pipeline(prompt, num_inference_steps=40, guidance_scale=8, height = 512, width = 712, scheduler = "DDIM").images[0]
|
| 35 |
image.save(f"img.png")
|
| 36 |
+
|
| 37 |
```
|
| 38 |
|
| 39 |
|