Update README.md
Browse files
README.md
CHANGED
|
@@ -59,7 +59,6 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
| 59 |
device_map="auto",
|
| 60 |
)
|
| 61 |
model = PeftModel.from_pretrained(model, adapter_name)
|
| 62 |
-
```
|
| 63 |
|
| 64 |
# Inference Example
|
| 65 |
messages = [
|
|
@@ -70,7 +69,9 @@ inputs = tokenizer.apply_chat_template(messages, tokenize=True, add_generation_p
|
|
| 70 |
# The model will output JSON immediately
|
| 71 |
outputs = model.generate(**inputs, max_new_tokens=512, temperature=0.1)
|
| 72 |
print(tokenizer.decode(outputs[0], skip_special_tokens=True))
|
| 73 |
-
|
|
|
|
| 74 |
Training data: u-10bei/structured_data_with_cot_dataset_512_v2
|
| 75 |
|
| 76 |
-
Dataset License: MIT License. This dataset is used and distributed under the terms of the MIT License.
|
|
|
|
|
|
| 59 |
device_map="auto",
|
| 60 |
)
|
| 61 |
model = PeftModel.from_pretrained(model, adapter_name)
|
|
|
|
| 62 |
|
| 63 |
# Inference Example
|
| 64 |
messages = [
|
|
|
|
| 69 |
# The model will output JSON immediately
|
| 70 |
outputs = model.generate(**inputs, max_new_tokens=512, temperature=0.1)
|
| 71 |
print(tokenizer.decode(outputs[0], skip_special_tokens=True))
|
| 72 |
+
```
|
| 73 |
+
## Sources & Terms (IMPORTANT)
|
| 74 |
Training data: u-10bei/structured_data_with_cot_dataset_512_v2
|
| 75 |
|
| 76 |
+
Dataset License: MIT License. This dataset is used and distributed under the terms of the MIT License.
|
| 77 |
+
Compliance: Users must comply with the MIT license (including copyright notice) and the base model's original terms of use.
|