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

Author: DenOfEquity

modules/dat_model.py

@@ -14,7 +14,7 @@ def __init__(self, user_path):
         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)

modules/esrgan_model.py

@@ -13,7 +13,7 @@ def __init__(self, dirname):
         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)