Merge remote-tracking branch 'myfork/img-rec' into img-rec-a

This commit is contained in:
BAIKEMARK
2025-06-13 10:30:34 +08:00
+5 -1
View File
@@ -34,10 +34,11 @@ config_changelog = """
[dependency-groups]
main = [
"llamafactory @ git+https://github.com/hiyouga/LLaMA-Factory.git",
"vllm==0.8.2; platform_system == 'Linux'",
"vllm==0.8.5.post1; platform_system == 'Linux'",
"torch>=2.6.0",
"transformers==4.52.1",
"accelerate==1.7.0",
"triton==3.2.0",
]
sparktts = [
"einops>=0.8.1",
@@ -62,6 +63,9 @@ conflicts = [
# [{ group = "wx" }, { group = "xcodec" }],
]
[tool.uv.pip]
torch-backend = "auto"
[tool.uv.sources]
torch = [
{ index = "pytorch-cu124", marker = "platform_system == 'Windows'" },