Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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-
|
| 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.")
|