Text Generation
Transformers
Safetensors
English
dami
linear-attention
long-context
initialization
ruler
Instructions to use SpeedAI-Lab/DAMI with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use SpeedAI-Lab/DAMI with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-generation", model="SpeedAI-Lab/DAMI")# pip install -U transformers accelerate # Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("SpeedAI-Lab/DAMI", device_map="auto") - Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- vLLM
How to use SpeedAI-Lab/DAMI with vLLM:
Install from pip and serve model
# Install vLLM from pip: pip install vllm # Start the vLLM server: vllm serve "SpeedAI-Lab/DAMI" # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:8000/v1/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "SpeedAI-Lab/DAMI", "prompt": "Once upon a time,", "max_tokens": 512, "temperature": 0.5 }'Use Docker
docker model run hf.co/SpeedAI-Lab/DAMI
- SGLang
How to use SpeedAI-Lab/DAMI with SGLang:
Install from pip and serve model
# Install SGLang from pip: pip install sglang # Start the SGLang server: python3 -m sglang.launch_server \ --model-path "SpeedAI-Lab/DAMI" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "SpeedAI-Lab/DAMI", "prompt": "Once upon a time,", "max_tokens": 512, "temperature": 0.5 }'Use Docker images
docker run --gpus all \ --shm-size 32g \ -p 30000:30000 \ -v ~/.cache/huggingface:/root/.cache/huggingface \ --env "HF_TOKEN=<secret>" \ --ipc=host \ lmsysorg/sglang:latest \ python3 -m sglang.launch_server \ --model-path "SpeedAI-Lab/DAMI" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "SpeedAI-Lab/DAMI", "prompt": "Once upon a time,", "max_tokens": 512, "temperature": 0.5 }' - Docker Model Runner
How to use SpeedAI-Lab/DAMI with Docker Model Runner:
docker model run hf.co/SpeedAI-Lab/DAMI
Synchronize architecture acknowledgements with the code repositories
Browse files
README.md
CHANGED
|
@@ -93,7 +93,7 @@ Average accuracy (%), as reported in the paper:
|
|
| 93 |
|
| 94 |
## Acknowledgement
|
| 95 |
|
| 96 |
-
|
| 97 |
|
| 98 |
Our retrieval evaluation uses [RULER](https://github.com/NVIDIA/RULER). Benchmark attribution and data notices are provided in [data/ruler/NOTICE.md](https://github.com/SpeedAI-Lab/DAMI/blob/main/data/ruler/NOTICE.md). We thank the authors and developers for making their work available. Please cite the corresponding papers when using these resources.
|
| 99 |
|
|
|
|
| 93 |
|
| 94 |
## Acknowledgement
|
| 95 |
|
| 96 |
+
We thank the authors and developers of [FLA](https://github.com/fla-org/flash-linear-attention), [RWKV](https://github.com/BlinkDL/RWKV-LM), [GatedDeltaNet](https://github.com/NVlabs/GatedDeltaNet), [GatedDeltaNet-2](https://github.com/NVlabs/GatedDeltaNet-2), [Kimi-Linear](https://github.com/MoonshotAI/Kimi-Linear), and [Mamba](https://github.com/state-spaces/mamba) for their contributions to the architectures and implementations used in this work. We also thank [Transformers](https://github.com/huggingface/transformers) and [FLAME](https://github.com/fla-org/flame) for their training infrastructure.
|
| 97 |
|
| 98 |
Our retrieval evaluation uses [RULER](https://github.com/NVIDIA/RULER). Benchmark attribution and data notices are provided in [data/ruler/NOTICE.md](https://github.com/SpeedAI-Lab/DAMI/blob/main/data/ruler/NOTICE.md). We thank the authors and developers for making their work available. Please cite the corresponding papers when using these resources.
|
| 99 |
|