Commit
•
58c6754
1
Parent(s):
5f4ccc9
Update handler.py
Browse files- handler.py +1 -0
handler.py
CHANGED
|
@@ -30,6 +30,7 @@ class EndpointHandler():
|
|
| 30 |
"""
|
| 31 |
inputs = data.pop("inputs", data)
|
| 32 |
random_generators= [torch.Generator().manual_seed(i) for i in range(inputs["num_images_per_prompt"])]
|
|
|
|
| 33 |
|
| 34 |
# run inference pipeline
|
| 35 |
with autocast(device.type):
|
|
|
|
| 30 |
"""
|
| 31 |
inputs = data.pop("inputs", data)
|
| 32 |
random_generators= [torch.Generator().manual_seed(i) for i in range(inputs["num_images_per_prompt"])]
|
| 33 |
+
print(random_generators)
|
| 34 |
|
| 35 |
# run inference pipeline
|
| 36 |
with autocast(device.type):
|