skynet/requirements.cuda.0.txt

scipy
triton
accelerate
transformers
huggingface_hub
diffusers[torch]>=0.18.0
invisible-watermark
torch==1.13.0+cu117
--extra-index-url https://download.pytorch.org/whl/cu117
First decupled architecture, still working on integrating tractor gpu workers 2022-12-10 21:18:03 +00:00			`scipy`
Add gpu worker tests to debug tractor lifecycle stuff also logging Minor tweak to cuda docker image Pin tractor branch Add triton to cuda reqs 2022-12-11 11:32:25 +00:00			`triton`
First decupled architecture, still working on integrating tractor gpu workers 2022-12-10 21:18:03 +00:00			`accelerate`
			`transformers`
			`huggingface_hub`
change stable2 model and how stablexl loads, add reqs 2023-07-19 21:25:28 +00:00			`diffusers[torch]>=0.18.0`
			`invisible-watermark`
First decupled architecture, still working on integrating tractor gpu workers 2022-12-10 21:18:03 +00:00			`torch==1.13.0+cu117`
			`--extra-index-url https://download.pytorch.org/whl/cu117`