Commit ·
03a2827
1
Parent(s): 026639f
Update
Browse files- handler.py +1 -1
handler.py
CHANGED
|
@@ -74,7 +74,7 @@ class EndpointHandler():
|
|
| 74 |
|
| 75 |
with torch.inference_mode():
|
| 76 |
prompt = "a photo of an astronaut riding a horse on mars"
|
| 77 |
-
image = pipe([prompt]*batch_size, num_inference_steps=20)
|
| 78 |
|
| 79 |
# image.save("astronaut_rides_horse.png")
|
| 80 |
|
|
|
|
| 74 |
|
| 75 |
with torch.inference_mode():
|
| 76 |
prompt = "a photo of an astronaut riding a horse on mars"
|
| 77 |
+
image = self.pipe([prompt]*batch_size, num_inference_steps=20)
|
| 78 |
|
| 79 |
# image.save("astronaut_rides_horse.png")
|
| 80 |
|