shouryashashank commited on
Commit
623e26d
1 Parent(s): 224cb5c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -70,7 +70,7 @@ The model `Precacons/ReasonGPT-2B-4bit` is a lightweight language model based on
70
  import predacons
71
 
72
  # Load the model and tokenizer
73
- model_path = "ReasonGPT-2B-4bit"
74
  model = predacons.load_model(model_path = model_path)
75
  tokenizer = predacons.load_tokenizer(model_path)
76
 
@@ -83,7 +83,7 @@ output,tokenizer =predacons.generate(model = model,
83
  trust_remote_code = True)
84
 
85
  # Decode and print the generated text
86
- generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
87
  print(generated_text)
88
  ```
89
 
 
70
  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
 
 
83
  trust_remote_code = True)
84
 
85
  # Decode and print the generated text
86
+ generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
87
  print(generated_text)
88
  ```
89