mirror of
https://github.com/deepseek-ai/DeepSeek-VL2.git
synced 2025-04-19 18:19:03 -04:00
Merge 8c69318fc7
into ef9f91e2b6
This commit is contained in:
commit
fa99a32471
@ -160,7 +160,7 @@ if __name__ == "__main__":
|
|||||||
default="deepseek-ai/deepseek-vl2",
|
default="deepseek-ai/deepseek-vl2",
|
||||||
help="model name or local path to the model")
|
help="model name or local path to the model")
|
||||||
parser.add_argument("--chunk_size", type=int, default=-1,
|
parser.add_argument("--chunk_size", type=int, default=-1,
|
||||||
help="chunk size for the model for prefiiling. "
|
help="chunk size for the model for prefilling. "
|
||||||
"When using 40G gpu for vl2-small, set a chunk_size for incremental_prefilling."
|
"When using 40G gpu for vl2-small, set a chunk_size for incremental_prefilling."
|
||||||
"Otherwise, default value is -1, which means we do not use incremental_prefilling.")
|
"Otherwise, default value is -1, which means we do not use incremental_prefilling.")
|
||||||
args = parser.parse_args()
|
args = parser.parse_args()
|
||||||
|
Loading…
Reference in New Issue
Block a user