AndranikSargsyan commited on
Commit
1c0f01f
·
1 Parent(s): a8a9bce

fix: Change GPU size to xlarge

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ IS_HF_SPACE = os.environ.get("SPACE_ID") is not None
31
 
32
  try:
33
  import spaces
34
- zero_gpu = spaces.GPU
35
  except ImportError:
36
  zero_gpu = lambda f: f
37
 
 
31
 
32
  try:
33
  import spaces
34
+ zero_gpu = spaces.GPU(size="xlarge")
35
  except ImportError:
36
  zero_gpu = lambda f: f
37