yuhuili commited on
Commit
d3e57a0
·
1 Parent(s): efd81c1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -187,10 +187,10 @@ parser = argparse.ArgumentParser()
187
  parser.add_argument(
188
  "--ea-model-path",
189
  type=str,
190
- default="lmsys/vicuna-7b-v1.3",
191
  help="The path to the weights. This can be a local folder or a Hugging Face repo ID.",
192
  )
193
- parser.add_argument("--base-model-path", type=str, default="/home/lyh/weights/hf/vicuna_v13/13B/",
194
  help="path of basemodel, huggingface project or local path")
195
  parser.add_argument(
196
  "--load-in-8bit", action="store_true", help="Use 8-bit quantization"
 
187
  parser.add_argument(
188
  "--ea-model-path",
189
  type=str,
190
+ default="yuhuili/EAGLE-Vicuna-7B-v1.3",
191
  help="The path to the weights. This can be a local folder or a Hugging Face repo ID.",
192
  )
193
+ parser.add_argument("--base-model-path", type=str, default="lmsys/vicuna-7b-v1.3",
194
  help="path of basemodel, huggingface project or local path")
195
  parser.add_argument(
196
  "--load-in-8bit", action="store_true", help="Use 8-bit quantization"