Spaces:

liruiw
/

hma

Running on Zero

hma / pre-requirements.txt

tune version

271be34 about 1 month ago

204 Bytes

	accelerate==0.30.1
	torchvision==0.17.0
	lpips==0.1.4
	matplotlib
	transformers==4.41.0
	torch==2.2.0
	lightning>2.3.1
	tqdm==4.66.4
	wandb
	xformers==0.0.25.post1
	git+https://github.com/janEbert/mup.git@fsdp-fix