Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -175,7 +175,7 @@ model_m = Qwen2_5_VLForConditionalGeneration.from_pretrained(
|
|
| 175 |
).to(device).eval()
|
| 176 |
|
| 177 |
|
| 178 |
-
@spaces.GPU
|
| 179 |
def generate_image(model_name: str, text: str, image: Image.Image,
|
| 180 |
max_new_tokens: int, temperature: float, top_p: float,
|
| 181 |
top_k: int, repetition_penalty: float):
|
|
|
|
| 175 |
).to(device).eval()
|
| 176 |
|
| 177 |
|
| 178 |
+
@spaces.GPU(duration=600)
|
| 179 |
def generate_image(model_name: str, text: str, image: Image.Image,
|
| 180 |
max_new_tokens: int, temperature: float, top_p: float,
|
| 181 |
top_k: int, repetition_penalty: float):
|