Skip to content

Latest commit

 

History

17 Commits

Folders and files

NameName
Last commit message
Last commit date
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

AutoTP

Description

AutoTP is a project for SungKyun Kim designed to automatically convert a single GPU huggingface model to a tensor-parallel model.

Installation

To get started with AutoTP, you'll need to install the necessary dependencies. Follow these steps:

  1. Clone the repository:

    git clone https://github.com/yourusername/AutoTP.git
    cd AutoTP
  2. Install the package:

    pip install .
  3. Install the required dependencies:

    pip install -r requirements.txt
  4. To run single GPU test, run the following command:

    cd benchmarks
    python test_single.sh
  5. To run distributed GPU test, run the following command:

    cd benchmarks
    python test_dist.sh

About

No description, website, or topics provided.

Resources

Stars

1 star

Watchers

1 watching

Forks

Releases

Packages

Contributors

Languages