diff --git a/requirements.txt b/requirements.txt index 5efaf80..d168bd4 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,5 +3,6 @@ flash_attn>=2.7.4.post1 tokenizers>=0.21.1 numpy>=2.1.0 rich -tabulate -typing_extensions \ No newline at end of file +typing_extensions +safetensors>=0.3.2 +ninja \ No newline at end of file diff --git a/setup.py b/setup.py index 523da57..1ee4580 100644 --- a/setup.py +++ b/setup.py @@ -86,21 +86,17 @@ setup( license="MIT", author="turboderp", install_requires=[ - "pandas", - "ninja", - "fastparquet", - "torch>=2.2.0", - "safetensors>=0.3.2", - "sentencepiece>=0.1.97", - "pygments", - "websockets", - "regex", - "numpy", + "torch>=2.6.0", + "flash_attn>=2.7.4.post1", + "tokenizers>=0.21.1", + "numpy>=2.1.0", "rich", - "pillow>=9.1.0" + "typing_extensions", + "ninja", + "safetensors>=0.3.2" ], include_package_data=True, - package_data={ + package_data = { "": ["py.typed"], }, verbose=verbose,