witcherderivia commited on
Commit
cd4aa10
·
verified ·
1 Parent(s): 83a5963

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -54,7 +54,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
54
  MAX_SEED = np.iinfo(np.int32).max
55
 
56
 
57
- @spaces.GPU
58
  def infer(
59
  content_ref,
60
  style_ref,
 
54
  MAX_SEED = np.iinfo(np.int32).max
55
 
56
 
57
+ @spaces.GPU(size="xlarge")
58
  def infer(
59
  content_ref,
60
  style_ref,