thefutureofai commited on
Commit
54cc1f4
·
verified ·
1 Parent(s): 9a47171

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -152,7 +152,7 @@ def load_model(model_path: str, device: str):
152
 
153
  def main():
154
  ap = argparse.ArgumentParser()
155
- ap.add_argument("--model_path", default="./outputs/phi-reasoning-crownelius-opus")
156
  ap.add_argument("--device", default="cuda:0")
157
  ap.add_argument("--system_prompt", default=DEFAULT_SYSTEM_PROMPT)
158
  ap.add_argument("--max_new_tokens", type=int, default=256)
 
152
 
153
  def main():
154
  ap = argparse.ArgumentParser()
155
+ ap.add_argument("--model_path", default="Aclevo/LBNET-2.7B-BASE")
156
  ap.add_argument("--device", default="cuda:0")
157
  ap.add_argument("--system_prompt", default=DEFAULT_SYSTEM_PROMPT)
158
  ap.add_argument("--max_new_tokens", type=int, default=256)