Skip to content

Adding load_in_8bit to inference_hf.py #756

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
Merged
merged 1 commit into from
Jul 17, 2023
Merged

Adding load_in_8bit to inference_hf.py #756

merged 1 commit into from
Jul 17, 2023

Conversation

airaria
Copy link
Contributor

@airaria airaria commented Jul 17, 2023 •
edited
Loading

Description

Adding --load_in_8bit option to inference_hf.py

Usage
Just add the --load_in_8bit to the launching command. For example:

python scripts/inference/inference_hf.py \
    --base_model path_to_merged_chinese_alpaca_plus \
    --with_prompt --interactive \
    --load_in_8bit

Related issue

#747

@airaria airaria requested a review from ymcui July 17, 2023 08:01
@ymcui ymcui merged commit 01eacec into main Jul 17, 2023
@ymcui ymcui deleted the update_inference_hf branch July 19, 2023 08:49
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@ymcui ymcui ymcui approved these changes

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@airaria @ymcui