layerdiffusion
commited on
Commit
•
2b769f6
1
Parent(s):
319b41c
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=
|
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)
|