deepseek-ai / deepseek-ai/DeepSeek-OCR
加载模型是否支持dtype="half"?
Open
- Dominant language
- Python
- Stars
- 23.9k
- Forks
- 2.2k
- PR merge metrics
- No merged PRs in 30d
Description
由于硬件设备不支持float16,尝试加载时进行修改
llm = LLM( model=MODEL_PATH, hf_overrides={"architectures": ["DeepseekOCRForCausalLM"]}, block_size=256, enforce_eager=False, trust_remote_code=True, max_model_len=8192, swap_space=0, max_num_seqs=MAX_CONCURRENCY, tensor_parallel_size=1, gpu_memory_utilization=0.9, disable_mm_preprocessor_cache=True, dtype="half" )
但是使用会存在相关报错,部分使用float,部分bfloat
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.