Spaces:
Running
Running
model update
Browse files- constants.py +9 -2
- frontend/webui/hf_demo.py +4 -4
constants.py
CHANGED
@@ -1,6 +1,8 @@
|
|
1 |
-
from os import environ
|
2 |
|
3 |
-
|
|
|
|
|
4 |
LCM_DEFAULT_MODEL = "stabilityai/sd-turbo"
|
5 |
LCM_DEFAULT_MODEL_OPENVINO = "rupeshs/sd-turbo-openvino"
|
6 |
APP_NAME = "FastSD CPU"
|
@@ -16,3 +18,8 @@ TAESDXL_MODEL = "madebyollin/taesdxl"
|
|
16 |
TAESD_MODEL_OPENVINO = "deinferno/taesd-openvino"
|
17 |
LCM_MODELS_FILE = "lcm-models.txt"
|
18 |
TAESDXL_MODEL_OPENVINO = "rupeshs/taesdxl-openvino"
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from os import environ, cpu_count
|
2 |
|
3 |
+
cpu_cores = cpu_count()
|
4 |
+
cpus = cpu_cores // 2 if cpu_cores else 0
|
5 |
+
APP_VERSION = "v1.0.0 beta 200"
|
6 |
LCM_DEFAULT_MODEL = "stabilityai/sd-turbo"
|
7 |
LCM_DEFAULT_MODEL_OPENVINO = "rupeshs/sd-turbo-openvino"
|
8 |
APP_NAME = "FastSD CPU"
|
|
|
18 |
TAESD_MODEL_OPENVINO = "deinferno/taesd-openvino"
|
19 |
LCM_MODELS_FILE = "lcm-models.txt"
|
20 |
TAESDXL_MODEL_OPENVINO = "rupeshs/taesdxl-openvino"
|
21 |
+
LORA_DIRECTORY = "lora_models"
|
22 |
+
CONTROLNET_DIRECTORY = "controlnet_models"
|
23 |
+
MODELS_DIRECTORY = "models"
|
24 |
+
GGUF_THREADS = environ.get("GGUF_THREADS", cpus)
|
25 |
+
TAEF1_MODEL_OPENVINO = "rupeshs/taef1-openvino"
|
frontend/webui/hf_demo.py
CHANGED
@@ -49,7 +49,7 @@ def predict(
|
|
49 |
lcm_diffusion_setting = LCMDiffusionSetting()
|
50 |
lcm_diffusion_setting.lcm_model_id = "rupeshs/hyper-sd-sdxl-1-step"
|
51 |
lcm_diffusion_setting.diffusion_task = DiffusionTask.text_to_image.value
|
52 |
-
lcm_diffusion_setting.openvino_lcm_model_id = "rupeshs/
|
53 |
lcm_diffusion_setting.use_lcm_lora = False
|
54 |
lcm_diffusion_setting.prompt = prompt
|
55 |
lcm_diffusion_setting.guidance_scale = 1.0
|
@@ -119,7 +119,7 @@ with gr.Blocks(css=css) as demo:
|
|
119 |
use_openvino = "" if is_openvino_device() else ""
|
120 |
gr.Markdown(
|
121 |
f"""# FastSD CPU demo {use_openvino}
|
122 |
-
**Device : {DEVICE.upper()} , {get_device_name()}**
|
123 |
""",
|
124 |
elem_id="intro",
|
125 |
)
|
@@ -148,9 +148,9 @@ with gr.Blocks(css=css) as demo:
|
|
148 |
with gr.Accordion("Advanced options", open=False):
|
149 |
steps = gr.Slider(
|
150 |
label="Steps",
|
151 |
-
value=
|
152 |
minimum=1,
|
153 |
-
maximum=
|
154 |
step=1,
|
155 |
)
|
156 |
seed = gr.Slider(
|
|
|
49 |
lcm_diffusion_setting = LCMDiffusionSetting()
|
50 |
lcm_diffusion_setting.lcm_model_id = "rupeshs/hyper-sd-sdxl-1-step"
|
51 |
lcm_diffusion_setting.diffusion_task = DiffusionTask.text_to_image.value
|
52 |
+
lcm_diffusion_setting.openvino_lcm_model_id = "rupeshs/LCM-dreamshaper-v7-openvino"
|
53 |
lcm_diffusion_setting.use_lcm_lora = False
|
54 |
lcm_diffusion_setting.prompt = prompt
|
55 |
lcm_diffusion_setting.guidance_scale = 1.0
|
|
|
119 |
use_openvino = "" if is_openvino_device() else ""
|
120 |
gr.Markdown(
|
121 |
f"""# FastSD CPU demo {use_openvino}
|
122 |
+
**Device : {DEVICE.upper()} , {get_device_name()} | OpenVINO **
|
123 |
""",
|
124 |
elem_id="intro",
|
125 |
)
|
|
|
148 |
with gr.Accordion("Advanced options", open=False):
|
149 |
steps = gr.Slider(
|
150 |
label="Steps",
|
151 |
+
value=3,
|
152 |
minimum=1,
|
153 |
+
maximum=4,
|
154 |
step=1,
|
155 |
)
|
156 |
seed = gr.Slider(
|