Skip to content

Commit

Permalink
make checkstyle
Browse files Browse the repository at this point in the history
  • Loading branch information
hebiao064 committed Dec 11, 2024
1 parent 41b97fa commit c84c56f
Showing 1 changed file with 5 additions and 7 deletions.
12 changes: 5 additions & 7 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,8 +2,10 @@

import subprocess
from typing import Literal

from setuptools import setup


def get_default_dependencies():
"""Determine the appropriate dependencies based on detected hardware."""
platform = get_platform()
Expand Down Expand Up @@ -42,19 +44,15 @@ def get_optional_dependencies():
"torchvision>=0.16.2",
"seaborn",
],
"transformers": [
"transformers~=4.0"
]
"transformers": ["transformers~=4.0"],
}
elif platform == "rocm":
return {
"dev": [
"setuptools-scm>=8",
"torchvision>=0.20.0.dev",
],
"transformers": [
"transformers>=4.44.2"
]
"transformers": ["transformers>=4.44.2"],
}


Expand Down Expand Up @@ -84,4 +82,4 @@ def get_platform() -> Literal["cuda", "rocm", "cpu"]:
packages=["liger_kernel"],
install_requires=get_default_dependencies(),
extras_require=get_optional_dependencies(),
)
)

0 comments on commit c84c56f

Please sign in to comment.