Skip to content

support combination of calibration datasets#70

Merged
wenhuach21 merged 20 commits into
mainfrom
local_customized_data
Apr 9, 2024
Merged

support combination of calibration datasets#70
wenhuach21 merged 20 commits into
mainfrom
local_customized_data

Conversation

@wenhuach21
Copy link
Copy Markdown
Contributor

@wenhuach21 wenhuach21 commented Apr 3, 2024

1 Supported local file for calibration
son and txt files are both supported

2 Supported Combination of different datasets

dataset = "./tmp.json"+ "," + "NeelNanda/pile-10k" + "," + "madao33/new-title-chinese" + "," + "mbpp"
bits, group_size, sym = 4, 128, True
autoround = AutoRound(
self.model,
self.tokenizer,
bits=bits,
group_size=group_size,
sym=sym,
dataset=dataset
)
autoround.quantize()

@wenhuach21 wenhuach21 changed the title support local json and txt file for calibration support combination and local calibration datasets Apr 3, 2024
@wenhuach21 wenhuach21 changed the title support combination and local calibration datasets support combination of calibration datasets Apr 3, 2024
@wenhuach21 wenhuach21 merged commit e8fb5da into main Apr 9, 2024
@wenhuach21 wenhuach21 deleted the local_customized_data branch April 9, 2024 08:27
attafosu pushed a commit to attafosu/auto-round that referenced this pull request Jul 23, 2024
Co-authored-by: Junyang Lin <justinlin930319@hotmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
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