Update README.md
Browse files
README.md
CHANGED
@@ -42,7 +42,7 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
42 |
trust_remote_code=True,
|
43 |
)
|
44 |
|
45 |
-
prompt = [{'role': 'user', 'content': '
|
46 |
inputs = tokenizer.apply_chat_template(
|
47 |
prompt,
|
48 |
add_generation_prompt=True,
|
@@ -143,6 +143,7 @@ The dataset is comprised of a mixture of open datasets large-scale datasets avai
|
|
143 |
- openchat/openchat_sharegpt4_dataset
|
144 |
- LDJnr/Capybara
|
145 |
- hkust-nlp/deita-10k-v0
|
|
|
146 |
|
147 |
2. Safety Datasets:
|
148 |
- Anthropic/hh-rlhf
|
|
|
42 |
trust_remote_code=True,
|
43 |
)
|
44 |
|
45 |
+
prompt = [{'role': 'user', 'content': 'Implement snake game using pygame'}]
|
46 |
inputs = tokenizer.apply_chat_template(
|
47 |
prompt,
|
48 |
add_generation_prompt=True,
|
|
|
143 |
- openchat/openchat_sharegpt4_dataset
|
144 |
- LDJnr/Capybara
|
145 |
- hkust-nlp/deita-10k-v0
|
146 |
+
- teknium/OpenHermes-2.5
|
147 |
|
148 |
2. Safety Datasets:
|
149 |
- Anthropic/hh-rlhf
|