Update README.md
Browse files
README.md
CHANGED
@@ -1,5 +1,6 @@
|
|
1 |
---
|
2 |
license: apache-2.0
|
|
|
3 |
datasets:
|
4 |
- bertin-project/bonanza-hf
|
5 |
- bertin-project/zenobia-instruct-hf
|
@@ -62,11 +63,8 @@ messages = [
|
|
62 |
]
|
63 |
|
64 |
generate_kwargs = {
|
65 |
-
|
66 |
"penalty_alpha": 0.6,
|
67 |
-
|
68 |
"max_new_tokens": 300,
|
69 |
-
|
70 |
}
|
71 |
pipe = pipeline("text-generation", model="bertin-project/Gromenauer-7B-Instruct", generate_kwargs=generate_kwargs)
|
72 |
pipe(messages)
|
|
|
1 |
---
|
2 |
license: apache-2.0
|
3 |
+
base_model: bertin-project/Gromenauer-7B
|
4 |
datasets:
|
5 |
- bertin-project/bonanza-hf
|
6 |
- bertin-project/zenobia-instruct-hf
|
|
|
63 |
]
|
64 |
|
65 |
generate_kwargs = {
|
|
|
66 |
"penalty_alpha": 0.6,
|
|
|
67 |
"max_new_tokens": 300,
|
|
|
68 |
}
|
69 |
pipe = pipeline("text-generation", model="bertin-project/Gromenauer-7B-Instruct", generate_kwargs=generate_kwargs)
|
70 |
pipe(messages)
|