Skip to content

CPU support#82

Open
esteininger wants to merge 1 commit intoDAMO-NLP-SG:mainfrom
esteininger:main-1
Open

CPU support#82
esteininger wants to merge 1 commit intoDAMO-NLP-SG:mainfrom
esteininger:main-1

Conversation

@esteininger
Copy link

swapped out CUDA params for CPU support

swapped out CUDA params for CPU support
@james-hu
Copy link

Hi @esteininger, I am new to this repo, so maybe this is a naive question, does it work with CPU? I mean, with this patch, is it able to generate a response in a reasonable time? And does it work for both training and inferencing?

BTW, I don't think the owner of this repo would accept this kind of functionality change. If it works with CPU, maybe making the behaviour controlled by an optional parameter specified somewhere (command line, or environment variable, or configuraiton file) is a better approach. Just my two cents.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants