filter lora files
Browse files
special/dl_display_results.py
CHANGED
@@ -165,7 +165,7 @@ display(HTML(CSS))
|
|
165 |
HR = widgets.HTML('<hr>')
|
166 |
|
167 |
header_widget = widgets.HTML(value='''
|
168 |
-
<div class="header_dl">DOWNLOAD RESULTS <span style="color: grey; opacity: 0.25;">| v0.2.
|
169 |
''')
|
170 |
|
171 |
# Adding content to "outputs_widgets_list"
|
@@ -185,7 +185,8 @@ vaes_widget = output_container_generator('VAEs', vaes_list).add_class("vae")
|
|
185 |
embeddings_list = [file for file in os.listdir(embeddings_dir) if not file.endswith('.txt') and not os.path.isdir(os.path.join(embeddings_dir, file))]
|
186 |
embeddings_widget = output_container_generator('Embeddings', embeddings_list).add_class("embedding")
|
187 |
# LoRAs
|
188 |
-
loras_list = [file for file in os.listdir(loras_dir) if not file.endswith('.txt')]
|
|
|
189 |
loras_widget = output_container_generator('LoRAs', loras_list).add_class("lora")
|
190 |
# Extensions
|
191 |
extensions_list = [folder for folder in os.listdir(extensions_dir) if os.path.isdir(os.path.join(extensions_dir, folder))]
|
|
|
165 |
HR = widgets.HTML('<hr>')
|
166 |
|
167 |
header_widget = widgets.HTML(value='''
|
168 |
+
<div class="header_dl">DOWNLOAD RESULTS <span style="color: grey; opacity: 0.25;">| v0.2.6</span></div>
|
169 |
''')
|
170 |
|
171 |
# Adding content to "outputs_widgets_list"
|
|
|
185 |
embeddings_list = [file for file in os.listdir(embeddings_dir) if not file.endswith('.txt') and not os.path.isdir(os.path.join(embeddings_dir, file))]
|
186 |
embeddings_widget = output_container_generator('Embeddings', embeddings_list).add_class("embedding")
|
187 |
# LoRAs
|
188 |
+
# loras_list = [file for file in os.listdir(loras_dir) if not file.endswith('.txt')]
|
189 |
+
loras_list = [file for file in os.listdir(loras_dir) if file.endswith('.safetensors')]
|
190 |
loras_widget = output_container_generator('LoRAs', loras_list).add_class("lora")
|
191 |
# Extensions
|
192 |
extensions_list = [folder for folder in os.listdir(extensions_dir) if os.path.isdir(os.path.join(extensions_dir, folder))]
|