NagisaNao commited on
Commit
fc0540a
·
verified ·
1 Parent(s): 876411e

filter lora files

Browse files
Files changed (1) hide show
  1. special/dl_display_results.py +3 -2
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.5</span></div>
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))]