huangrh9 commited on
Commit
58219ef
·
verified ·
1 Parent(s): 1a9b7e3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -898,7 +898,7 @@ def build_demo(embed_mode):
898
  if __name__ == "__main__":
899
  parser = argparse.ArgumentParser()
900
  # --- Add arguments for ILLUME configs and checkpoints ---
901
- parser.add_argument("--model_name", type=str, default="ILLUME-MLLM/illume_plus-qwen-2_5-3b-hf",
902
  help="Name for builder.")
903
  parser.add_argument("--torch_dtype", type=str, default='fp32', choices=['fp32', 'bf16', 'fp16'],
904
  help="Computation data type.")
 
898
  if __name__ == "__main__":
899
  parser = argparse.ArgumentParser()
900
  # --- Add arguments for ILLUME configs and checkpoints ---
901
+ parser.add_argument("--model_name", type=str, default="ILLUME-MLLM/illume_plus-qwen2_5-3b-hf",
902
  help="Name for builder.")
903
  parser.add_argument("--torch_dtype", type=str, default='fp32', choices=['fp32', 'bf16', 'fp16'],
904
  help="Computation data type.")