PR #1826 Use device as context manager for init_on_device

Use device as context manager for init_on_device #1826

sgugger merged 1 commit into huggingface:main from shingjan:shingjan/use_device

shingjan force pushed from 2b414f8c to c16b9920 2 years ago

shingjan force pushed from c16b9920 to b488347f 2 years ago

shingjan force pushed from b488347f to ece578c2 2 years ago

sgugger commented on 2023-08-09

BenjaminBossan commented on 2023-08-09

shingjan force pushed from ece578c2 to c6d746dc 2 years ago

shingjan force pushed from c6d746dc to 38f49b45 2 years ago

shingjan force pushed from 38f49b45 to 4afad012 2 years ago

shingjan force pushed from 4afad012 to 7237f9a3 2 years ago

shingjan requested a review from

BenjaminBossan 2 years ago

shingjan requested a review from

sgugger 2 years ago

use device as context manager for init_on_device

32c3dbb7

shingjan force pushed from 7237f9a3 to 32c3dbb7 2 years ago

sgugger approved these changes on 2023-08-10

sgugger merged 058a3546 into main 2 years ago

shingjan deleted the shingjan/use_device branch 2 years ago

Reviewers

sgugger

BenjaminBossan

Assignees

No one assigned

Labels

None yet

Milestone

No milestone