tjellm commited on
Commit
d4f6685
1 Parent(s): dd0929d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +22 -1
README.md CHANGED
@@ -13,4 +13,25 @@ tags:
13
  - DirectML
14
  - DML
15
  - conversational
16
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
13
  - DirectML
14
  - DML
15
  - conversational
16
+ ---
17
+
18
+ # Llama-2-7b-chat ONNX models for DirectML
19
+ This repository hosts the optimized versions of [meta-llama/Llama-2-7b-chat-hf](https://huggingface.co/meta-llama/Llama-2-7b-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-7b-chat-int4-onnx-directml --local-dir .\llama-2-7b-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-7b-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.