Update handler.py
Browse files- handler.py +9 -9
handler.py
CHANGED
|
@@ -50,15 +50,15 @@ class EndpointHandler:
|
|
| 50 |
generator = torch.manual_seed(seed)
|
| 51 |
self.record+=1
|
| 52 |
start_time = time.time()
|
| 53 |
-
time.sleep(
|
| 54 |
-
result = self.pipe( # type: ignore
|
| 55 |
-
|
| 56 |
-
|
| 57 |
-
|
| 58 |
-
|
| 59 |
-
|
| 60 |
-
|
| 61 |
-
).images[0]
|
| 62 |
end_time = time.time()
|
| 63 |
time_taken = end_time - start_time
|
| 64 |
print(f"Time taken: {time_taken:.2f} seconds")
|
|
|
|
| 50 |
generator = torch.manual_seed(seed)
|
| 51 |
self.record+=1
|
| 52 |
start_time = time.time()
|
| 53 |
+
time.sleep(6)
|
| 54 |
+
# result = self.pipe( # type: ignore
|
| 55 |
+
# prompt,
|
| 56 |
+
# height=height,
|
| 57 |
+
# width=width,
|
| 58 |
+
# guidance_scale=guidance_scale,
|
| 59 |
+
# num_inference_steps=num_inference_steps,
|
| 60 |
+
# generator=generator,
|
| 61 |
+
# ).images[0]
|
| 62 |
end_time = time.time()
|
| 63 |
time_taken = end_time - start_time
|
| 64 |
print(f"Time taken: {time_taken:.2f} seconds")
|