Skip to content

Commit 22e9119

Browse files
committed
entrypoint: add snake diffusion
1 parent 17a61aa commit 22e9119

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

dweam/utils/entrypoint.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -76,6 +76,15 @@
7676
branch="csgo",
7777
),
7878
],
79+
"snake_diffusion": [
80+
PathSource(
81+
path=Path("snake-diffusion"),
82+
),
83+
GitBranchSource(
84+
git="https://github.com/dweam-team/snake-diffusion",
85+
branch="main",
86+
),
87+
],
7988
"diamond_atari": [
8089
PathSource(
8190
path=Path("diamond"),
@@ -128,6 +137,7 @@ def install_game_source(log: BoundLogger, venv_path: Path, source: GameSource, n
128137
pip_base_args = [
129138
str(pip_path),
130139
"install",
140+
# "--force-reinstall",
131141
"--extra-index-url",
132142
"https://download.pytorch.org/whl/cu121",
133143
]

0 commit comments

Comments
 (0)