deepspeed-model-memory-usage

Running

deepspeed-model-memory-usage / requirements.txt

Update requirements.txt

f4fbb52 almost 2 years ago

129 Bytes

	accelerate @ git+https://github.com/huggingface/accelerate
	transformers
	timm
	huggingface_hub
	tabulate
	einops
	flash-attn --no-deps