aswain4 commited on
Commit
c96af93
·
verified ·
1 Parent(s): b9c82fe

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +11 -6
README.md CHANGED
@@ -35,6 +35,17 @@ The goal of this model is to improve the quality and efficiency of code generati
35
  - **Repository:** [mistralai/Mistral-7B-v0.1](https://huggingface.co/mistralai/Mistral-7B-v0.1}{mistralai/Mistral-7B-v0.1)
36
  - **Paper:** [Mistral-7B](https://arxiv.org/abs/2310.06825)
37
 
 
 
 
 
 
 
 
 
 
 
 
38
  ## Uses
39
 
40
  <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
@@ -63,12 +74,6 @@ The model was trained on a dataset that is predominantly Python code; therefore,
63
 
64
  Users (both direct and downstream) should be made aware of the risks and limitations of the model. Please read the above section before using this model.
65
 
66
- ## How to Get Started with the Model
67
-
68
- Use the code below to get started with the model.
69
-
70
- [More Information Needed]
71
-
72
  ## Training Details
73
 
74
  ### Training Data
 
35
  - **Repository:** [mistralai/Mistral-7B-v0.1](https://huggingface.co/mistralai/Mistral-7B-v0.1}{mistralai/Mistral-7B-v0.1)
36
  - **Paper:** [Mistral-7B](https://arxiv.org/abs/2310.06825)
37
 
38
+ ## How to Get Started with the Model
39
+
40
+ Use the code below to get started with the model.
41
+
42
+ ```python
43
+ from transformers import AutoTokenizer, AutoModelForCausalLM
44
+
45
+ tokenizer = AutoTokenizer.from_pretrained('mistralai/Mistral-7B-v0.1')
46
+ model = AutoModelForCausalLM.from_pretrained('aswain4/custom_coding_LLM', device_map='auto', torch_dtype=torch.bfloat16)
47
+ ```
48
+
49
  ## Uses
50
 
51
  <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
 
74
 
75
  Users (both direct and downstream) should be made aware of the risks and limitations of the model. Please read the above section before using this model.
76
 
 
 
 
 
 
 
77
  ## Training Details
78
 
79
  ### Training Data