remove deepfloyd if

ddPn08 · Jul 7, 2023 · 5c1280e · 5c1280e
1 parent 8c4c847
commit 5c1280e
Show file tree

Hide file tree

Showing 5 changed files with 3 additions and 450 deletions.
diff --git a/modules/config.py b/modules/config.py
@@ -45,7 +45,6 @@
 
 parser.add_argument("--xformers", action="store_true")
 parser.add_argument("--tensorrt", action="store_true")
-parser.add_argument("--deepfloyd_if", action="store_true")
 
 cmd_opts, _ = parser.parse_known_args(
     os.environ["COMMANDLINE_ARGS"].split(" ")

diff --git a/modules/diffusion/pipelines/deepfloyd_if.py b/modules/diffusion/pipelines/deepfloyd_if.py
diff --git a/modules/model_manager.py b/modules/model_manager.py
@@ -12,9 +12,7 @@
 
 sd_models: List[DiffusersModel] = []
 sd_model: Optional[DiffusersModel] = None
-mode: Literal["stable-diffusion", "deepfloyd_if"] = (
-    "deepfloyd_if" if config.get("deepfloyd_if") else "stable-diffusion"
-)
+mode: Literal["stable-diffusion"] = "stable-diffusion"
 
 
 def get_model(model_id: str):