lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-30 03:01:15 +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 = [] self.scalers = []
super().__init__() 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) name = modelloader.friendly_name(file)
scaler_data = UpscalerData(name, file, upscaler=self, scale=None) scaler_data = UpscalerData(name, file, upscaler=self, scale=None)
self.scalers.append(scaler_data) self.scalers.append(scaler_data)

2
modules/esrgan_model.py
View File

@@ -13,7 +13,7 @@ class UpscalerESRGAN(Upscaler):
self.scalers = [] self.scalers = []
self.user_path = dirname self.user_path = dirname
super().__init__() super().__init__()
model_paths = self.find_models(ext_filter=[".pt", ".pth"]) model_paths = self.find_models(ext_filter=[".pt", ".pth", ".safetensors"])
scalers = [] scalers = []
if len(model_paths) == 0: if len(model_paths) == 0:
scaler_data = UpscalerData(self.model_name, self.model_url, self, 4) scaler_data = UpscalerData(self.model_name, self.model_url, self, 4)