shouryashashank
commited on
Commit
•
9264fd8
1
Parent(s):
623e26d
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 |
|