entrypoint: add snake diffusion

dweam-team · Jan 18, 2025 · 22e9119 · 22e9119
1 parent 17a61aa
commit 22e9119
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/dweam/utils/entrypoint.py b/dweam/utils/entrypoint.py
@@ -76,6 +76,15 @@
                 branch="csgo",
             ),
         ],
+        "snake_diffusion": [
+            PathSource(
+                path=Path("snake-diffusion"),
+            ),
+            GitBranchSource(
+                git="https://github.com/dweam-team/snake-diffusion",
+                branch="main",
+            ),
+        ],
         "diamond_atari": [
             PathSource(
                 path=Path("diamond"),
@@ -128,6 +137,7 @@ def install_game_source(log: BoundLogger, venv_path: Path, source: GameSource, n
         pip_base_args = [
             str(pip_path),
             "install",
+            # "--force-reinstall",
             "--extra-index-url",
             "https://download.pytorch.org/whl/cu121",
         ]