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 |
|