layerdiffusion commited on
Commit
2b769f6
1 Parent(s): 319b41c
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -120,7 +120,7 @@ def resize_without_crop(image, target_width, target_height):
120
  return np.array(resized_image)
121
 
122
 
123
- @spaces.GPU(duration=180)
124
  @torch.inference_mode()
125
  def chat_fn(message: str, history: list, seed:int, temperature: float, top_p: float, max_new_tokens: int) -> str:
126
  print('Chat begin:', message)
 
120
  return np.array(resized_image)
121
 
122
 
123
+ @spaces.GPU(duration=120)
124
  @torch.inference_mode()
125
  def chat_fn(message: str, history: list, seed:int, temperature: float, top_p: float, max_new_tokens: int) -> str:
126
  print('Chat begin:', message)