skynet/requirements.cuda.txt

torch
scipy
triton
xformers
accelerate
transformers
huggingface_hub
diffusers[torch]
invisible-watermark

basicsr
realesrgan

--extra-index-url https://download.pytorch.org/whl/cu117
Enable certain memory optmization options on cases where trying to load a large model on a low end card 2023-08-14 03:15:45 +00:00			`torch`
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`
Enable certain memory optmization options on cases where trying to load a large model on a low end card 2023-08-14 03:15:45 +00:00			`xformers`
First decupled architecture, still working on integrating tractor gpu workers 2022-12-10 21:18:03 +00:00			`accelerate`
			`transformers`
			`huggingface_hub`
Enable certain memory optmization options on cases where trying to load a large model on a low end card 2023-08-14 03:15:45 +00:00			`diffusers[torch]`
change stable2 model and how stablexl loads, add reqs 2023-07-19 21:25:28 +00:00			`invisible-watermark`
Enable certain memory optmization options on cases where trying to load a large model on a low end card 2023-08-14 03:15:45 +00:00
			`basicsr`
			`realesrgan`

First decupled architecture, still working on integrating tractor gpu workers 2022-12-10 21:18:03 +00:00			`--extra-index-url https://download.pytorch.org/whl/cu117`