fix(models): pin Transformers device_map to GPU 0 when auto sharding breaks sampling#44
Open
Aafiya-H wants to merge 3 commits into
Open
fix(models): pin Transformers device_map to GPU 0 when auto sharding breaks sampling#44Aafiya-H wants to merge 3 commits into
Aafiya-H wants to merge 3 commits into