Update README.md
Browse files
README.md
CHANGED
|
@@ -71,14 +71,14 @@ import predacons
|
|
| 71 |
|
| 72 |
# Load the model and tokenizer
|
| 73 |
model_path = "Precacons/ReasonGPT-2B-4bit"
|
| 74 |
-
model = predacons.load_model(model_path
|
| 75 |
tokenizer = predacons.load_tokenizer(model_path)
|
| 76 |
|
| 77 |
# Example usage
|
| 78 |
sequence = "Explain the concept of acceleration in physics."
|
| 79 |
output,tokenizer =predacons.generate(model = model,
|
| 80 |
sequence = sequence,
|
| 81 |
-
max_length =
|
| 82 |
tokenizer = tokenizer,
|
| 83 |
trust_remote_code = True)
|
| 84 |
|
|
|
|
| 71 |
|
| 72 |
# Load the model and tokenizer
|
| 73 |
model_path = "Precacons/ReasonGPT-2B-4bit"
|
| 74 |
+
model = predacons.load_model(model_path)
|
| 75 |
tokenizer = predacons.load_tokenizer(model_path)
|
| 76 |
|
| 77 |
# Example usage
|
| 78 |
sequence = "Explain the concept of acceleration in physics."
|
| 79 |
output,tokenizer =predacons.generate(model = model,
|
| 80 |
sequence = sequence,
|
| 81 |
+
max_length = 500,
|
| 82 |
tokenizer = tokenizer,
|
| 83 |
trust_remote_code = True)
|
| 84 |
|