Update README.md
Browse files
README.md
CHANGED
@@ -65,6 +65,7 @@ model = model.to('cuda', dtype=torch.bfloat16)
|
|
65 |
processor = AutoProcessor.from_pretrained(
|
66 |
model_args.model_name,
|
67 |
trust_remote_code=True,
|
|
|
68 |
num_crops=4,
|
69 |
)
|
70 |
|
|
|
65 |
processor = AutoProcessor.from_pretrained(
|
66 |
model_args.model_name,
|
67 |
trust_remote_code=True,
|
68 |
+
model_backbone='phi3_v',
|
69 |
num_crops=4,
|
70 |
)
|
71 |
|