Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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=
|
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=
|
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,
|