ByteForge commited on
Commit
5e7482c
1 Parent(s): 4c2817c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +9 -3
README.md CHANGED
@@ -32,9 +32,15 @@ Deepseek-Coder-7B-Instruct-v1.5 is continue pre-trained from Deepseek-LLM 7B on
32
  Here give some examples of how to use our model.
33
  #### Chat Model Inference
34
  ```python
35
- from transformers import AutoTokenizer, AutoModelForCausalLM
36
- tokenizer = AutoTokenizer.from_pretrained("deepseek-ai/deepseek-coder-7b-instruct-v1.5", trust_remote_code=True)
37
- model = AutoModelForCausalLM.from_pretrained("deepseek-ai/deepseek-coder-7b-instruct-v1.5", trust_remote_code=True).cuda()
 
 
 
 
 
 
38
  messages=[
39
  { 'role': 'user', 'content': "write a quick sort algorithm in python."}
40
  ]
 
32
  Here give some examples of how to use our model.
33
  #### Chat Model Inference
34
  ```python
35
+ import ctranslate2
36
+ import transformers
37
+
38
+ from huggingface_hub import snapshot_download
39
+ model_id = "ByteForge/DS-7b-1.5_Instruct-ct2-int8_float32"
40
+ model_path = snapshot_download(model_id)
41
+ model = ctranslate2.Generator(model_path, device='cuda')
42
+ tokenizer = transformers.AutoTokenizer.from_pretrained(model_id)
43
+
44
  messages=[
45
  { 'role': 'user', 'content': "write a quick sort algorithm in python."}
46
  ]