Point to main branch of turbine-models in requirements.

nod-ai · Jul 15, 2024 · aa4acef · aa4acef
1 parent d5013fd
commit aa4acef
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -8,7 +8,7 @@ wheel
 
 torch==2.3.0
 shark-turbine @ git+https://github.com/iree-org/iree-turbine.git@main
-turbine-models @ git+https://github.com/nod-ai/SHARK-Turbine.git@ean-unify-sd#subdirectory=models
+turbine-models @ git+https://github.com/nod-ai/SHARK-Turbine.git@main#subdirectory=models
 diffusers @ git+https://github.com/nod-ai/[email protected]
 brevitas @ git+https://github.com/Xilinx/brevitas.git@6695e8df7f6a2c7715b9ed69c4b78157376bb60b