tahamajs commited on
Commit
952a443
·
verified ·
1 Parent(s): 5da53e7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +42 -162
README.md CHANGED
@@ -2,199 +2,79 @@
2
  library_name: transformers
3
  tags:
4
  - unsloth
 
 
 
 
5
  ---
6
 
7
- # Model Card for Model ID
8
-
9
- <!-- Provide a quick summary of what the model is/does. -->
10
-
11
 
 
12
 
13
  ## Model Details
14
 
15
  ### Model Description
16
 
17
- <!-- Provide a longer summary of what this model is. -->
18
-
19
- This is the model card of a 🤗 transformers model that has been pushed on the Hub. This model card has been automatically generated.
20
 
21
- - **Developed by:** [More Information Needed]
22
- - **Funded by [optional]:** [More Information Needed]
23
- - **Shared by [optional]:** [More Information Needed]
24
- - **Model type:** [More Information Needed]
25
- - **Language(s) (NLP):** [More Information Needed]
26
- - **License:** [More Information Needed]
27
- - **Finetuned from model [optional]:** [More Information Needed]
28
 
29
- ### Model Sources [optional]
30
 
31
- <!-- Provide the basic links for the model. -->
32
-
33
- - **Repository:** [More Information Needed]
34
- - **Paper [optional]:** [More Information Needed]
35
- - **Demo [optional]:** [More Information Needed]
36
 
37
  ## Uses
38
 
39
- <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
40
-
41
  ### Direct Use
42
 
43
- <!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
 
 
 
44
 
45
- [More Information Needed]
46
 
47
- ### Downstream Use [optional]
48
-
49
- <!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
50
-
51
- [More Information Needed]
52
 
53
  ### Out-of-Scope Use
54
 
55
- <!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
56
-
57
- [More Information Needed]
58
 
59
  ## Bias, Risks, and Limitations
60
 
61
- <!-- This section is meant to convey both technical and sociotechnical limitations. -->
62
-
63
- [More Information Needed]
64
 
65
  ### Recommendations
66
 
67
- <!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
68
-
69
- Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
 
70
 
71
  ## How to Get Started with the Model
72
 
73
- Use the code below to get started with the model.
74
-
75
- [More Information Needed]
76
-
77
- ## Training Details
78
-
79
- ### Training Data
80
-
81
- <!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
82
-
83
- [More Information Needed]
84
-
85
- ### Training Procedure
86
-
87
- <!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
88
-
89
- #### Preprocessing [optional]
90
-
91
- [More Information Needed]
92
-
93
-
94
- #### Training Hyperparameters
95
-
96
- - **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
97
-
98
- #### Speeds, Sizes, Times [optional]
99
-
100
- <!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
101
-
102
- [More Information Needed]
103
-
104
- ## Evaluation
105
-
106
- <!-- This section describes the evaluation protocols and provides the results. -->
107
-
108
- ### Testing Data, Factors & Metrics
109
-
110
- #### Testing Data
111
-
112
- <!-- This should link to a Dataset Card if possible. -->
113
-
114
- [More Information Needed]
115
-
116
- #### Factors
117
-
118
- <!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
119
-
120
- [More Information Needed]
121
-
122
- #### Metrics
123
-
124
- <!-- These are the evaluation metrics being used, ideally with a description of why. -->
125
-
126
- [More Information Needed]
127
-
128
- ### Results
129
-
130
- [More Information Needed]
131
-
132
- #### Summary
133
-
134
-
135
-
136
- ## Model Examination [optional]
137
-
138
- <!-- Relevant interpretability work for the model goes here -->
139
-
140
- [More Information Needed]
141
-
142
- ## Environmental Impact
143
-
144
- <!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
145
-
146
- Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
147
-
148
- - **Hardware Type:** [More Information Needed]
149
- - **Hours used:** [More Information Needed]
150
- - **Cloud Provider:** [More Information Needed]
151
- - **Compute Region:** [More Information Needed]
152
- - **Carbon Emitted:** [More Information Needed]
153
-
154
- ## Technical Specifications [optional]
155
-
156
- ### Model Architecture and Objective
157
-
158
- [More Information Needed]
159
-
160
- ### Compute Infrastructure
161
-
162
- [More Information Needed]
163
-
164
- #### Hardware
165
-
166
- [More Information Needed]
167
-
168
- #### Software
169
-
170
- [More Information Needed]
171
-
172
- ## Citation [optional]
173
-
174
- <!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
175
-
176
- **BibTeX:**
177
-
178
- [More Information Needed]
179
-
180
- **APA:**
181
-
182
- [More Information Needed]
183
-
184
- ## Glossary [optional]
185
-
186
- <!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
187
-
188
- [More Information Needed]
189
-
190
- ## More Information [optional]
191
-
192
- [More Information Needed]
193
 
194
- ## Model Card Authors [optional]
 
195
 
196
- [More Information Needed]
 
197
 
198
- ## Model Card Contact
 
 
 
199
 
200
- [More Information Needed]
 
2
  library_name: transformers
3
  tags:
4
  - unsloth
5
+ - dpo
6
+ - orpo
7
+ - lora
8
+ - preference-optimization
9
  ---
10
 
11
+ # Model Card for Llama-3.2-3B ORPO Fine-Tuned Model with LoRA
 
 
 
12
 
13
+ This model is a fine-tuned version of the base model **unsloth/Llama-3.2-3B-Instruct-bnb-4bit** using Odds Ratio Preference Optimization (ORPO) with LoRA-based adaptation. The training leverages a dataset of pairwise (chosen vs. rejected) responses to align the model with human preferences without the need for a separate reward or reference model.
14
 
15
  ## Model Details
16
 
17
  ### Model Description
18
 
19
+ This is a fine-tuned language model that has been optimized using ORPO—a direct preference optimization method that eliminates the need for a reference model. The base model, **unsloth/Llama-3.2-3B-Instruct-bnb-4bit**, is adapted using Low-Rank Adaptation (LoRA) with a rank and alpha of 64, allowing for efficient fine-tuning with only a small fraction of the model's parameters updated. The fine-tuning is performed on a dataset consisting of approximately 1,600 examples (sampled from "mlabonne/orpo-dpo-mix-40k"), where the model learns to favor the "chosen" response over the "rejected" one directly through odds ratio optimization.
 
 
20
 
21
+ - **Developed by:** [Your Name or Organization]
22
+ - **Model Type:** Causal Language Model (Instruction-Finetuned)
23
+ - **Base Model:** unsloth/Llama-3.2-3B-Instruct-bnb-4bit
24
+ - **Training Method:** ORPO (Odds Ratio Preference Optimization) with LoRA
25
+ - **Quantization:** 4-bit
26
+ - **Language:** English (primarily)
27
+ - **License:** [Specify License, e.g., Apache-2.0]
28
 
29
+ ### Model Sources
30
 
31
+ - **Repository:** [Link to the repository on Hugging Face]
32
+ - **Paper:** [Reference any paper if available, or "N/A"]
33
+ - **Demo:** [Link to a demo if available]
 
 
34
 
35
  ## Uses
36
 
 
 
37
  ### Direct Use
38
 
39
+ This model is intended for tasks that benefit from preference-aligned generation, such as:
40
+ - Instruction following
41
+ - Chatbot response generation
42
+ - Content creation where human-aligned quality is crucial
43
 
44
+ ### Downstream Use
45
 
46
+ This model can be further fine-tuned or adapted for domain-specific applications where human preferences play a significant role in output quality.
 
 
 
 
47
 
48
  ### Out-of-Scope Use
49
 
50
+ - Applications requiring rigorous factual correctness (e.g., medical or legal advice) without further domain-specific fine-tuning.
51
+ - Use cases involving sensitive content where model biases could lead to harmful outcomes.
 
52
 
53
  ## Bias, Risks, and Limitations
54
 
55
+ - **Bias:** The model may still exhibit biases inherited from the base model and the fine-tuning data.
56
+ - **Risks:** Users should be cautious in applications where incorrect or biased information could have serious consequences.
57
+ - **Limitations:** As a fine-tuned model using preference optimization, its performance is tied to the quality and diversity of the training data. It may not generalize well to contexts significantly different from its training set.
58
 
59
  ### Recommendations
60
 
61
+ Users should:
62
+ - Evaluate the model on their specific use case.
63
+ - Monitor outputs for potential bias or factual inaccuracies.
64
+ - Fine-tune further if necessary to better align with specific requirements.
65
 
66
  ## How to Get Started with the Model
67
 
68
+ Below is an example code snippet to load and use the model:
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
69
 
70
+ ```python
71
+ from transformers import AutoModelForCausalLM, AutoTokenizer
72
 
73
+ model = AutoModelForCausalLM.from_pretrained("your-username/llama-3.2-3b-orpo-lora64")
74
+ tokenizer = AutoTokenizer.from_pretrained("your-username/llama-3.2-3b-orpo-lora64")
75
 
76
+ input_text = "Please explain the benefits of using ORPO for fine-tuning language models."
77
+ inputs = tokenizer(input_text, return_tensors="pt").to("cuda")
78
+ outputs = model.generate(**inputs, max_new_tokens=100)
79
+ print(tokenizer.decode(outputs[0]))
80