Allow Safetensors Upscalers

AUTOMATIC1111 · Jan 31, 2024 · 26fe815 · 26fe815
1 parent 9651f8c
commit 26fe815
Show file tree

Hide file tree

Showing 8 changed files with 8 additions and 8 deletions.
diff --git a/modules/compact_model.py b/modules/compact_model.py
@@ -11,7 +11,7 @@ def __init__(self, dirname):
         self.scalers = []
         self.user_path = dirname
         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)
             scale = None
             scaler_data = UpscalerData(name, file, upscaler=self, scale=scale)

diff --git a/modules/dat_model.py b/modules/dat_model.py
@@ -11,7 +11,7 @@ def __init__(self, dirname):
         self.scalers = []
         self.user_path = dirname
         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)
             scale = None
             scaler_data = UpscalerData(name, file, upscaler=self, scale=scale)

diff --git a/modules/esrgan_model.py b/modules/esrgan_model.py
@@ -11,7 +11,7 @@ def __init__(self, dirname):
         self.scalers = []
         self.user_path = dirname
         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)
             scale = None
             scaler_data = UpscalerData(name, file, upscaler=self, scale=scale)

diff --git a/modules/grl_model.py b/modules/grl_model.py
@@ -11,7 +11,7 @@ def __init__(self, dirname):
         self.scalers = []
         self.user_path = dirname
         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)
             scale = None
             scaler_data = UpscalerData(name, file, upscaler=self, scale=scale)

diff --git a/modules/hat_model.py b/modules/hat_model.py
@@ -11,7 +11,7 @@ def __init__(self, dirname):
         self.scalers = []
         self.user_path = dirname
         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)
             scale = None
             scaler_data = UpscalerData(name, file, upscaler=self, scale=scale)

diff --git a/modules/omnisr_model.py b/modules/omnisr_model.py
@@ -11,7 +11,7 @@ def __init__(self, dirname):
         self.scalers = []
         self.user_path = dirname
         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)
             scale = None
             scaler_data = UpscalerData(name, file, upscaler=self, scale=scale)

diff --git a/modules/span_model.py b/modules/span_model.py
@@ -11,7 +11,7 @@ def __init__(self, dirname):
         self.scalers = []
         self.user_path = dirname
         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)
             scale = None
             scaler_data = UpscalerData(name, file, upscaler=self, scale=scale)

diff --git a/modules/srformer_model.py b/modules/srformer_model.py
@@ -11,7 +11,7 @@ def __init__(self, dirname):
         self.scalers = []
         self.user_path = dirname
         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)
             scale = None
             scaler_data = UpscalerData(name, file, upscaler=self, scale=scale)