Skip to content
Snippets Groups Projects
Unverified Commit e1c7674b authored by Mashiro's avatar Mashiro Committed by GitHub
Browse files

[Fix] Fix potential OOM error when loading checkpoint with `map_location="cuda:0"` (#665) 

* [Fix] Change the default value of map_location in PretrainedInit to cpu

* Fix as comment
parent 618a063d
No related branches found
No related tags found
Hide whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment

Datenschutz und Impressum