Update README.md
Browse files
README.md
CHANGED
|
@@ -62,9 +62,6 @@ from llava.train.dataset import tokenizer_image_token
|
|
| 62 |
|
| 63 |
|
| 64 |
if __name__ == "__main__":
|
| 65 |
-
parser = transformers.HfArgumentParser(
|
| 66 |
-
(ModelArguments, DataArguments, TrainingArguments))
|
| 67 |
-
model_args, data_args, training_args = parser.parse_args_into_dataclasses()
|
| 68 |
model_path = 'toshi456/llava-jp-1.3b-v1.1'
|
| 69 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 70 |
torch_dtype = torch.bfloat16 if device=="cuda" else torch.float32
|
|
|
|
| 62 |
|
| 63 |
|
| 64 |
if __name__ == "__main__":
|
|
|
|
|
|
|
|
|
|
| 65 |
model_path = 'toshi456/llava-jp-1.3b-v1.1'
|
| 66 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 67 |
torch_dtype = torch.bfloat16 if device=="cuda" else torch.float32
|