yuhuili commited on
Commit
1aba41b
1 Parent(s): af81f25

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -241,10 +241,10 @@ parser = argparse.ArgumentParser()
241
  parser.add_argument(
242
  "--ea-model-path",
243
  type=str,
244
- default="8B",
245
  help="The path to the weights. This can be a local folder or a Hugging Face repo ID.",
246
  )
247
- parser.add_argument("--base-model-path", type=str, default="meta-llama/Meta-Llama-3-8B-Instruct",
248
  help="path of basemodel, huggingface project or local path")
249
  parser.add_argument(
250
  "--load-in-8bit", action="store_true", help="Use 8-bit quantization"
 
241
  parser.add_argument(
242
  "--ea-model-path",
243
  type=str,
244
+ default="yuhuili/EAGLE-LLaMA3-Instruct-8B",
245
  help="The path to the weights. This can be a local folder or a Hugging Face repo ID.",
246
  )
247
+ parser.add_argument("--base-model-path", type=str, default="8B",
248
  help="path of basemodel, huggingface project or local path")
249
  parser.add_argument(
250
  "--load-in-8bit", action="store_true", help="Use 8-bit quantization"