rupeshs commited on
Commit
083a73a
·
1 Parent(s): c08d7f7

model update

Browse files
Files changed (2) hide show
  1. constants.py +9 -2
  2. frontend/webui/hf_demo.py +4 -4
constants.py CHANGED
@@ -1,6 +1,8 @@
1
- from os import environ
2
 
3
- APP_VERSION = "v1.0.0 beta 23"
 
 
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/sd-turbo-openvino"
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=1,
152
  minimum=1,
153
- maximum=3,
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(