Update README.md
Browse files
README.md
CHANGED
|
@@ -80,9 +80,9 @@ pipe = pipeline(
|
|
| 80 |
"text-generation",
|
| 81 |
model="Bertug1911/BrtGPT-1-Pre",
|
| 82 |
trust_remote_code=True,
|
| 83 |
-
top_k=40, #
|
| 84 |
-
temperature=0.8, #
|
| 85 |
-
max_new_tokens=
|
| 86 |
)
|
| 87 |
|
| 88 |
# Messages
|
|
|
|
| 80 |
"text-generation",
|
| 81 |
model="Bertug1911/BrtGPT-1-Pre",
|
| 82 |
trust_remote_code=True,
|
| 83 |
+
top_k=40, # Good for creativity
|
| 84 |
+
temperature=0.8, # Good for creativity
|
| 85 |
+
max_new_tokens=128 # Default maximum model output (Maximum 1024)
|
| 86 |
)
|
| 87 |
|
| 88 |
# Messages
|