Add ".pt" to the Lora extensions (#1518)

https://discord.com/channels/1014774730907209781/1014774732018683926/1144179143873929288 There seem to be ".pt" LORA files in the wild.
2023-08-24 12:39:42 +02:00 · 2023-08-24 12:39:42 +02:00 · 31edce4a60
commit 31edce4a60
parent 1b6aae9678
1 changed files with 1 additions and 1 deletions
--- 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"],