FilesExpand file tree

main

/

setup.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

16 lines (15 loc) · 407 Bytes

main

/

setup.py

Copy path

Top

File metadata and controls

• Code
• Blame

16 lines (15 loc) · 407 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

from setuptools import setup, find_packages

setup(

name="transformers-patch",

version="0.1.0",

author="Xingkai Yu",

author_email="xingkai@deepseek.com",

description="patches for huggingface transformers to save memory",

packages=find_packages(),

install_requires=[

"torch>=2.4",

"triton>=3.0",

"transformers>=4.51.0"

],

python_requires=">=3.9",

)