refoundd commited on
Commit
012cf5d
·
verified ·
1 Parent(s): a8d6202

Update handler.py

Browse files
Files changed (1) hide show
  1. 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(10)
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")
 
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")