From 2fc58ad93580ec739bd3d9db9a1f11010af8686e Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Thu, 17 Apr 2025 23:10:43 -0700
Subject: [PATCH] Consider files with .pt extension in the new model menu
 function

---
 modules/utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/modules/utils.py b/modules/utils.py
index 81cd085a..f6be7541 100644
--- a/modules/utils.py
+++ b/modules/utils.py
@@ -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