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(
|