aredden / flux-fp8-api

Flux diffusion model implementation using quantized fp8 matmul & remaining layers use faster half precision accumulate, which is ~2x faster on consumer devices.
Apache License 2.0
109 stars 12 forks source link

python 3.10 compatibility #2

Closed XmYx closed 3 weeks ago

XmYx commented 3 weeks ago

StrEnum was introduced in python3.11, this allows the package to run on 3.10

aredden commented 3 weeks ago

Ah, this seems like a good idea. I'll merge.