prithivMLmods commited on
Commit
eab92bd
·
verified ·
1 Parent(s): 8a317aa

update app

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -165,7 +165,7 @@ def update_dimensions_on_upload(image):
165
 
166
  return new_width, new_height
167
 
168
- @spaces.GPU
169
  def infer(
170
  images,
171
  prompt,
@@ -254,7 +254,7 @@ def infer(
254
  gc.collect()
255
  torch.cuda.empty_cache()
256
 
257
- @spaces.GPU
258
  def infer_example(images, prompt, lora_adapter):
259
  if not images:
260
  return None, 0
 
165
 
166
  return new_width, new_height
167
 
168
+ @spaces.GPU(size="xlarge")
169
  def infer(
170
  images,
171
  prompt,
 
254
  gc.collect()
255
  torch.cuda.empty_cache()
256
 
257
+ @spaces.GPU(size="xlarge")
258
  def infer_example(images, prompt, lora_adapter):
259
  if not images:
260
  return None, 0