cleanup for fp8 and requirements etc

This commit is contained in:
Ashwin Bharambe 2024-07-20 23:21:41 -07:00
parent 2428701951
commit d73fed5cc3
9 changed files with 78 additions and 905 deletions

31
fp8_requirements.txt Normal file
View file

@ -0,0 +1,31 @@
--extra-index-url https://download.pytorch.org/whl/nightly/cu121
torch>=2.4.0.dev20240531,<2.4.1
accelerate
black==24.4.2
codeshield
fairscale
fastapi
fire
flake8
huggingface-hub
httpx
hydra-core
hydra-zen
json-strong-typing
matplotlib
omegaconf
pandas
Pillow
pre-commit
pydantic==1.10.13
pydantic_core==2.18.2
python-dotenv
python-openapi
requests
tiktoken
transformers
ufmt==2.7.0
usort==1.0.8
uvicorn
zmq
fbgemm-gpu==0.8.0rc4