transformers
[DeepSeekV2] Fix integration tests OOM: use device_map=auto instead of 8-bit quantization
#47991
Merged

[DeepSeekV2] Fix integration tests OOM: use device_map=auto instead of 8-bit quantization #47991

ydshieh merged 4 commits into main from fix_deepseek_v2_oom
ydshieh
ydshieh [DeepSeekV2] Fix integration tests: use device_map=auto, remove 8-bit…
6c8a0c1f
ydshieh [DeepSeekV2] Update expected values for device_map=auto bfloat16 loading
85583b2e
ydshieh [DeepSeekV2] Fix curly apostrophe in batch_fa2 expected text, add tea…
4b37e451
ydshieh
github-actions
HuggingFaceDocBuilderDev
github-actions
ydshieh [DeepSeekV2] Round expected logit values to 4 decimal places
e8a32a53
ydshieh
github-actions
ydshieh
github-actions
github-actions
github-actions
ydshieh ydshieh merged a61d5f9e into main 2 days ago
ydshieh ydshieh deleted the fix_deepseek_v2_oom branch 2 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
Milestone