Add Flux.1 S and Flux.1 D filtering capabilities (#333)

* Add Flux.1 and Flux.1 D filtering capabilities

* Update civitai_api.py

* Update civitai_gui.py

* Update civitai_api.py

Realized I went too complicated about it, no changes to civitai_api.py are necessary

* Update civitai_gui.py

Realized that all was necessary was properly inputting the right spelling of the base models

* Update civitai_api.py

* Update civitai_gui.py
pull/348/head
ArchAngelAries 2024-10-10 14:15:03 -06:00 committed by GitHub
parent aa32745e64
commit 47a6204122
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 3 additions and 3 deletions

View File

@ -1214,4 +1214,4 @@ def api_error_msg(input_string):
elif input_string == "no_items":
return div + "Failed to retrieve any models from CivitAI<br>The servers might be too busy or down if the issue persists."
else:
return div + "The CivitAI-API failed to respond due to an error.<br>Check the logs for more details."
return div + "The CivitAI-API failed to respond due to an error.<br>Check the logs for more details."

View File

@ -188,7 +188,7 @@ def on_ui_tabs():
with gr.Row():
content_type = gr.Dropdown(label='Content type:', choices=content_choices, value=None, type="value", multiselect=True, elem_id="centerText")
with gr.Row():
base_filter = gr.Dropdown(label='Base model:', multiselect=True, choices=["SD 1.4","SD 1.5","SD 1.5 LCM","SD 2.0","SD 2.0 768","SD 2.1","SD 2.1 768","SD 2.1 Unclip","SDXL 0.9","SDXL 1.0","SDXL 1.0 LCM","SDXL Distilled","SDXL Turbo","SDXL Lightning","Stable Cascade","Pony","SVD","SVD XT","Playground v2","PixArt a","Other"], value=None, type="value", elem_id="centerText")
base_filter = gr.Dropdown(label='Base model:', multiselect=True, choices=["SD 1.4","SD 1.5","SD 1.5 LCM","SD 2.0","SD 2.0 768","SD 2.1","SD 2.1 768","SD 2.1 Unclip","SDXL 0.9","SDXL 1.0","SDXL 1.0 LCM","SDXL Distilled","SDXL Turbo","SDXL Lightning","Stable Cascade","Pony","SVD","SVD XT","Playground v2","PixArt a", "Flux.1 S", "Flux.1 D","Other"], value=None, type="value", elem_id="centerText")
with gr.Row():
period_type = gr.Dropdown(label='Time period:', choices=["All Time", "Year", "Month", "Week", "Day"], value="All Time", type="value", elem_id="centerText")
sort_type = gr.Dropdown(label='Sort by:', choices=["Newest","Oldest","Most Downloaded","Highest Rated","Most Liked","Most Buzz","Most Discussed","Most Collected","Most Images"], value="Most Downloaded", type="value", elem_id="centerText")
@ -1412,4 +1412,4 @@ def on_ui_settings():
shared.opts.add_option(f"{setting_name}_subfolder", shared.OptionInfo("None", folder_name, gr.Dropdown, make_lambda(folder, desc), section=download, **({'category_id': cat_id} if ver_bool else {})))
script_callbacks.on_ui_tabs(on_ui_tabs)
script_callbacks.on_ui_settings(on_ui_settings)
script_callbacks.on_ui_settings(on_ui_settings)