Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
gmlwns2000 authored Jun 25, 2024
1 parent 5300d32 commit 9026688
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,8 @@

After installation, you can access to `hip` package from any project. `hip` is code name of HiP attention.

[vLLM Implementation](https://github.com/DeepAuto-AI/vllm)

## TL;DR

**We provide OpenAI Compatible API server built with vLLM and HiP attention!**
Expand Down Expand Up @@ -284,4 +286,4 @@ python examples/openai_chat_image_client.py --image-file="images/cherry_blossom.
## Example training command
```bash
OMP_NUM_THREADS=16 CUDA_VISIBLE_DEVICES=0,1,2,3 PYTHONPATH=. accelerate launch --num_processes=4 --main_process_port 29501 hip/trainer/hip_trainer_hf.py --method hip --block_size_q 32 --block_size_k 2 --k 512 --lora_r 256 --dataset openwebtext --dense_layers 4 --name bs16_warmup10_dq2k --dense_queries 2048 --seq_len 32768 --disable_kd --sparsity_reg 0.01 --gradient_accumulation_steps 4 --warmup_steps 10 --model giraffe13b --using_deepspeed
```
```

0 comments on commit 9026688

Please sign in to comment.