Switch dependency branch to main

Signed-off-by: Kacper Dąbrowski <[email protected]>
RobotecAI · Sep 13, 2024 · 0f2249c · 0f2249c
1 parent 86ea1b2
commit 0f2249c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -8,7 +8,7 @@ readme = "README.md"
 [tool.poetry.dependencies]
 python = "^3.10, <3.13"
 torch = "^2.3"
-diffusion-policy = {git = "https://github.com/RobotecAI/diffusion_policy.git", branch = "kd/bump_torch"}
+diffusion-policy = {git = "https://github.com/RobotecAI/diffusion_policy.git", branch = "main"}
 diffusers = "^0.29.1"
 efficientnet-pytorch = "^0.7.1"
 vit-pytorch = "^1.7.0"