Skip to content
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.

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

fix device map #148

Merged
merged 1 commit into from
Jun 2, 2023
Merged

fix device map #148

merged 1 commit into from
Jun 2, 2023

Conversation

winglian
Copy link
Collaborator

@winglian winglian commented Jun 2, 2023

causes the device to be {'':'cuda:0'}. symptom is failure to train with this error:

ValueError: You can't train a model that has been loaded in 8-bit precision on a different device than the one you're training on. Make sure you loaded the model on the corr
ect device using for example `device_map={'':torch.cuda.current_device()}you're training on. Make sure you loaded the model on the correct device using for example `device_m
ap={'':torch.cuda.current_device() or device_map={'':torch.xpu.current_device()}                                                                                             

@winglian winglian merged commit 4ab0c8b into main Jun 2, 2023
@winglian winglian deleted the fix-device-load branch June 2, 2023 18:37
mkeoliya pushed a commit to mkeoliya/axolotl that referenced this pull request Dec 15, 2023
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.

1 participant