diff --git a/modules/dat_model.py b/modules/dat_model.py index 75b7afd1..c60d9de3 100644 --- a/modules/dat_model.py +++ b/modules/dat_model.py @@ -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) diff --git a/modules/esrgan_model.py b/modules/esrgan_model.py index 20d47b00..394fe241 100644 --- a/modules/esrgan_model.py +++ b/modules/esrgan_model.py @@ -13,7 +13,7 @@ class UpscalerESRGAN(Upscaler): self.scalers = [] self.user_path = dirname super().__init__() - model_paths = self.find_models(ext_filter=[".pt", ".pth"]) + model_paths = self.find_models(ext_filter=[".pt", ".pth", ".safetensors"]) scalers = [] if len(model_paths) == 0: scaler_data = UpscalerData(self.model_name, self.model_url, self, 4)