shouryashashank commited on
Commit
9264fd8
1 Parent(s): 623e26d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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 = 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 = 50,
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