We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
No description provided.
The text was updated successfully, but these errors were encountered:
建议使用vllm启动vllm-project/vllm#4650
Sorry, something went wrong.
HuggingFace代码中accelerate库对模型的显存分配计算有问题,目前示例代码已修改,预计大幅缩短模型加载速度。
accelerate
加载模型的代码修改为:
model = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True, device_map="sequential", torch_dtype=torch.bfloat16, max_memory=max_memory, attn_implementation="eager")
No branches or pull requests
No description provided.
The text was updated successfully, but these errors were encountered: