revise for safesentors file

AIRI-AUBE · Mar 21, 2023 · 0231ec4 · 0231ec4
1 parent d427458
commit 0231ec4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/webui.py b/webui.py
@@ -237,7 +237,7 @@ def webui():
                 api_endpoint = os.environ['api_endpoint']
                 endpoint_name = os.environ['endpoint_name']
                 for file in os.listdir(sd_models_path):
-                    if os.path.isfile(os.path.join(sd_models_path, file)) and file.endswith('.ckpt'):
+                    if os.path.isfile(os.path.join(sd_models_path, file)) and (file.endswith('.ckpt') or file.endswith('.ckpt')):
                         hash = modules.sd_models.model_hash(os.path.join(sd_models_path, file))
                         item = {}
                         item['model_name'] = file