oobabooga/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2026-01-25 10:00:50 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Consider files with .pt extension in the new model menu function

Browse source
This commit is contained in:
oobabooga 2025-04-17 23:10:43 -07:00
parent d78abe480b
commit 2fc58ad935
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
modules/utils.py
View file

@ -91,7 +91,7 @@ def get_available_models():
item_path = model_dir / item
if item_path.is_dir():
# Check if there are safetensors files directly under this directory
if any(file.lower().endswith('.safetensors') for file in os.listdir(item_path) if (item_path / file).is_file()):
if any(file.lower().endswith(('.safetensors', '.pt')) for file in os.listdir(item_path) if (item_path / file).is_file()):
dirs_with_safetensors.add(item)
# Find valid model directories