Update README.md
Browse files
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="
|
| 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)
|