tjellm commited on
Commit
e1fe368
1 Parent(s): 3e1bc15

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +37 -0
README.md ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: llama2
3
+ language:
4
+ - en
5
+ pipeline_tag: text-generation
6
+ tags:
7
+ - facebook
8
+ - meta
9
+ - pytorch
10
+ - llama
11
+ - llama-2
12
+ - ONNX
13
+ - DirectML
14
+ - DML
15
+ - conversational
16
+ ---
17
+
18
+ # Llama-2-13b-chat ONNX models for DirectML
19
+ This repository hosts the optimized versions of [meta-llama/Llama-2-13b-chat-hf](https://huggingface.co/meta-llama/Llama-2-13b-chat-hf) to accelerate inference with ONNX Runtime for DirectML.
20
+
21
+ ## Usage on Windows (Intel / AMD / Nvidia / Qualcomm)
22
+ ```powershell
23
+ conda create -n onnx python=3.10
24
+ conda activate onnx
25
+ winget install -e --id GitHub.GitLFS
26
+ pip install huggingface-hub[cli]
27
+ huggingface-cli download EmbeddedLLM/llama-2-13b-chat-int4-onnx-directml --local-dir .\llama-2-13b-chat
28
+ pip install numpy
29
+ Invoke-WebRequest -Uri "https://raw.githubusercontent.com/microsoft/onnxruntime-genai/main/examples/python/phi3-qa.py" -OutFile "phi3-qa.py"
30
+ pip install onnxruntime-directml
31
+ pip install --pre onnxruntime-genai-directml
32
+ conda install conda-forge::vs2015_runtime
33
+ python phi3-qa.py -m .\llama-2-13b-chat
34
+ ```
35
+
36
+ ## What is DirectML
37
+ DirectML is a high-performance, hardware-accelerated DirectX 12 library for machine learning. DirectML provides GPU acceleration for common machine learning tasks across a broad range of supported hardware and drivers, including all DirectX 12-capable GPUs from vendors such as AMD, Intel, NVIDIA, and Qualcomm.