Spaces:

multimodalart
/

ai-toolkit

Running on CPU Upgrade

multimodalart HF Staff commited on Sep 17

Commit

c3dc017

verified ·

1 Parent(s): a265f7c

Update ui/src/app/api/hf-jobs/route.ts

Files changed (1) hide show

ui/src/app/api/hf-jobs/route.ts CHANGED Viewed

@@ -93,6 +93,7 @@ function generateUVScript({ jobConfig, datasetRepo, namespace, token }: {
 # dependencies = [
 #     "torch>=2.0.0",
 #     "torchvision",
 #     "torchao==0.10.0",
 #     "safetensors",
 #     "diffusers @ git+https://github.com/huggingface/diffusers@7a2b78bf0f788d311cc96b61e660a8e13e3b1e63",

 # dependencies = [
 #     "torch>=2.0.0",
 #     "torchvision",
+#     "torchaudio",
 #     "torchao==0.10.0",
 #     "safetensors",
 #     "diffusers @ git+https://github.com/huggingface/diffusers@7a2b78bf0f788d311cc96b61e660a8e13e3b1e63",