Skip to content

Commit da24c2f

Browse files
committed
Update CUDA version to 13 and remove specific version constraints for several Python packages in cog.yaml for improved flexibility in dependency management.
1 parent 448b5bd commit da24c2f

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

cog.yaml

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
build:
55
# set to true if your model requires a GPU
66
gpu: true
7-
cuda: "12.1"
7+
cuda: "13"
88

99
# python version in the form '3.11' or '3.11.4'
1010
python_version: "3.11"
@@ -16,18 +16,18 @@ build:
1616

1717
# a list of packages in the format <package-name>==<version>
1818
python_packages:
19-
- diffusers==0.35.1
19+
- diffusers
2020
- transformers
2121
- sentencepiece
2222
- protobuf
23-
- huggingface-hub==0.34.0
23+
- huggingface-hub
2424
- peft
2525
- opencv-python
2626
- lightning
2727
- datasets
2828
- prodigyopt
2929
- wandb
30-
- Pillow==9.5.0
30+
- Pillow
3131
- accelerate
3232
- py_real_esrgan
3333
- rembg

0 commit comments

Comments
 (0)