Update app.py
Browse files
app.py
CHANGED
|
@@ -59,6 +59,7 @@ def log_gpu_memory():
|
|
| 59 |
# --------- MinDalle Image Generation Functions ---------
|
| 60 |
# Load MinDalle model once
|
| 61 |
# Dynamically determine device and precision
|
|
|
|
| 62 |
def load_min_dalle_model(models_root: str = 'pretrained'):
|
| 63 |
"""
|
| 64 |
Load the MinDalle model, automatically selecting device and precision.
|
|
|
|
| 59 |
# --------- MinDalle Image Generation Functions ---------
|
| 60 |
# Load MinDalle model once
|
| 61 |
# Dynamically determine device and precision
|
| 62 |
+
@spaces.GPU(duration=60 * 3)
|
| 63 |
def load_min_dalle_model(models_root: str = 'pretrained'):
|
| 64 |
"""
|
| 65 |
Load the MinDalle model, automatically selecting device and precision.
|