LPX55 commited on
Commit
fddb0bb
·
verified ·
1 Parent(s): af9721e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -41,7 +41,7 @@ for p in glob.glob(f"{cp_dir}/segment-anything-2/*.yaml"):
41
  shutil.copy(p, "sam2_configs")
42
 
43
  from modelmod import FluxModel
44
- model = FluxModel(device=DEVICE, is_turbo=False, required_features=['controlnet', 'depth'], is_quantization=False) # , 'sam'
45
 
46
  QWEN2VLFLUX_MODES = ["variation", "img2img", "inpaint", "controlnet", "controlnet-inpaint"]
47
  QWEN2VLFLUX_ASPECT_RATIO = ["1:1", "16:9", "9:16", "2.4:1", "3:4", "4:3"]
@@ -104,7 +104,7 @@ def fetch_from_url(url: str, name: str):
104
  print(e)
105
  return None
106
 
107
- @spaces.GPU(duration=100)
108
  @torch.inference_mode()
109
  def process(
110
  mode: str,
 
41
  shutil.copy(p, "sam2_configs")
42
 
43
  from modelmod import FluxModel
44
+ model = FluxModel(device=DEVICE, is_turbo=False, required_features=['controlnet', 'depth'], is_quantization=True) # , 'sam'
45
 
46
  QWEN2VLFLUX_MODES = ["variation", "img2img", "inpaint", "controlnet", "controlnet-inpaint"]
47
  QWEN2VLFLUX_ASPECT_RATIO = ["1:1", "16:9", "9:16", "2.4:1", "3:4", "4:3"]
 
104
  print(e)
105
  return None
106
 
107
+ @spaces.GPU(duration=60)
108
  @torch.inference_mode()
109
  def process(
110
  mode: str,