transformers
559af445 - Cast byte_count to int in caching_allocator_warmup for MPS compatibility (#43608)

Commit

30 days ago

Cast byte_count to int in caching_allocator_warmup for MPS compatibility (#43608) Fix: Cast byte_count to int in caching_allocator_warmup for MPS compatibility

References

#43608 - Cast byte_count to int in caching_allocator_warmup for MPS compatibility

Author

tobyliu2004

Parents

02943590

transformers 559af445 - Cast byte_count to int in caching_allocator_warmup for MPS compatibility (#43608)

transformers
559af445 - Cast byte_count to int in caching_allocator_warmup for MPS compatibility (#43608)