Spaces:
Running
Running
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.
|