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"],