lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-03-11 07:59:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

add .safetensors to known extensions for DAT and ESRGAN (#2651)

Browse Source
This commit is contained in:
DenOfEquity
2025-02-13 16:16:18 +00:00
committed by GitHub
parent a9e247eb2e
commit ade69764a1
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
modules/dat_model.py
View File

@@ -14,7 +14,7 @@ class UpscalerDAT(Upscaler):
self.scalers = []
super().__init__()
for file in self.find_models(ext_filter=[".pt", ".pth"]):
for file in self.find_models(ext_filter=[".pt", ".pth", ".safetensors"]):
name = modelloader.friendly_name(file)
scaler_data = UpscalerData(name, file, upscaler=self, scale=None)
self.scalers.append(scaler_data)