Skip to content

Support loading F8_E4M3 from safetensors - #1859

Merged
barronalex merged 1 commit into
mainfrom
f8_e4m3
Feb 14, 2025
Merged

Support loading F8_E4M3 from safetensors#1859
barronalex merged 1 commit into
mainfrom
f8_e4m3

Conversation

@barronalex

Copy link
Copy Markdown
Contributor

Loads F8_E4M3 tensors in bf16 from safetensors.

The load only works on the GPU at the moment because it uses a custom kernel.

@angeloskath angeloskath left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looks great!

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.

2 participants