toshi456
/

llava-jp-1.3b-v1.1

text-generation

image-captioning

Inference Endpoints

Model card Files Files and versions Community

toshi456 commited on Apr 20

Commit

4b4e71a

•

1 Parent(s): 251d8d9

Update README.md

Files changed (1) hide show

README.md +0 -3

README.md CHANGED Viewed

@@ -62,9 +62,6 @@ from llava.train.dataset import tokenizer_image_token
 if __name__ == "__main__":
-    parser = transformers.HfArgumentParser(
-        (ModelArguments, DataArguments, TrainingArguments))
-    model_args, data_args, training_args = parser.parse_args_into_dataclasses()
     model_path = 'toshi456/llava-jp-1.3b-v1.1'
     device = "cuda" if torch.cuda.is_available() else "cpu"
     torch_dtype = torch.bfloat16 if device=="cuda" else torch.float32

 if __name__ == "__main__":
     model_path = 'toshi456/llava-jp-1.3b-v1.1'
     device = "cuda" if torch.cuda.is_available() else "cpu"
     torch_dtype = torch.bfloat16 if device=="cuda" else torch.float32