Update README.md
Browse files
README.md
CHANGED
|
@@ -122,7 +122,7 @@ from transformers import AutoModelForCausalLM, AutoProcessor
|
|
| 122 |
from qwen_vl_utils import process_vision_info
|
| 123 |
|
| 124 |
# Load Model
|
| 125 |
-
model_path = "
|
| 126 |
model = AutoModelForCausalLM.from_pretrained(
|
| 127 |
model_path,
|
| 128 |
torch_dtype=torch.bfloat16,
|
|
|
|
| 122 |
from qwen_vl_utils import process_vision_info
|
| 123 |
|
| 124 |
# Load Model
|
| 125 |
+
model_path = "hustvl/InfiniteVL" # Replace with your HF repo ID
|
| 126 |
model = AutoModelForCausalLM.from_pretrained(
|
| 127 |
model_path,
|
| 128 |
torch_dtype=torch.bfloat16,
|