Skip to content

gpu choose #563

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

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

Jump to bottom
Open
1panzi opened this issue May 16, 2025 · 2 comments
Open

gpu choose #563

1panzi opened this issue May 16, 2025 · 2 comments
Labels
question Further information is requested

Comments

@1panzi
Copy link

1panzi commented May 16, 2025

Describe the Question

你好 这个库 加载模型时，可以选择gpu吗 我这里运行爆outof memoy
@1panzi 1panzi added the question Further information is requested label May 16, 2025
@shibing624
Copy link
Owner

可以用gpu，out of memory是内存不够，跑小点儿的模型试试。具体看readme和源码。

@1panzi
Copy link
Author

1panzi commented May 16, 2025

我使用GTP加载的7b模型 我有4卡3090 估计默认加的第一个卡 现在我用的
cuda_device=1 args={"fp16":True} 目前因该没问题
GptCorrector(model_name_or_path=model_path,cuda_device=cuda_device,args=args)

但是bert的浅浅的看了下 貌似不好控制

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
question Further information is requested
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@shibing624 @1panzi