6 Commits

Author	SHA1	Message	Date
Deyao Zhu	3c9d321b79	fix an import bug in image_text_pair_builder.py	2023-04-17 16:57:20 +03:00
Deyao Zhu	b76d5c5a50	add argument to switch 8bit	2023-04-17 16:54:00 +03:00
152334H	700f05d079	set num_beam default to 1 instead of completely removing it	2023-04-17 14:56:18 +08:00
152334H	6a8e6e49ff	load in 8bit and run ViT-L in cpu to reduce vram use	2023-04-17 13:52:09 +08:00
152334H	017fa43d03	disable beam search to reduce vram use	2023-04-17 13:51:39 +08:00
Deyao Zhu	f1a33af227	first commit	2023-04-17 01:04:16 +03:00