From 31edce4a60bf68840a5ffff0775bc488612c3320 Mon Sep 17 00:00:00 2001
From: JeLuF <jf@mormo.org>
Date: Thu, 24 Aug 2023 12:39:42 +0200
Subject: [PATCH] Add ".pt" to the Lora extensions (#1518)

https://discord.com/channels/1014774730907209781/1014774732018683926/1144179143873929288

There seem to be ".pt" LORA files in the wild.
---
 ui/easydiffusion/model_manager.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ui/easydiffusion/model_manager.py b/ui/easydiffusion/model_manager.py
index 7848291a..d88694e2 100644
--- a/ui/easydiffusion/model_manager.py
+++ b/ui/easydiffusion/model_manager.py
@@ -29,7 +29,7 @@ MODEL_EXTENSIONS = {
     "hypernetwork": [".pt", ".safetensors"],
     "gfpgan": [".pth"],
     "realesrgan": [".pth"],
-    "lora": [".ckpt", ".safetensors"],
+    "lora": [".ckpt", ".safetensors", ".pt"],
     "codeformer": [".pth"],
     "embeddings": [".pt", ".bin", ".safetensors"],
     "controlnet": [".pth", ".safetensors"],