shilinxu
/

InternVL2_5-8B

Model card Files Files and versions Metrics Training metrics Community

shilinxu commited on about 1 month ago

Commit

1a3cfbf

·

verified ·

1 Parent(s): 5ba7fec

Update README.md

Files changed (1) hide show

README.md +2 -0

README.md CHANGED Viewed

@@ -92,6 +92,8 @@ model = AutoModel.from_pretrained(
     low_cpu_mem_usage=True,
     use_flash_attn=False,
     trust_remote_code=True).eval().cuda()
 tokenizer = AutoTokenizer.from_pretrained(path, trust_remote_code=True, use_fast=False)
 # set the max number of tiles in `max_num`

     low_cpu_mem_usage=True,
     use_flash_attn=False,
     trust_remote_code=True).eval().cuda()
+model.num_image_token = 150
 tokenizer = AutoTokenizer.from_pretrained(path, trust_remote_code=True, use_fast=False)
 # set the max number of tiles in `max_num`