ruslanmv commited on
Commit
bb51163
·
verified ·
1 Parent(s): 0036d0a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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.