chengzeyi / stable-fast

Best inference performance optimization framework for HuggingFace Diffusers on NVIDIA GPUs.
MIT License
1.16k stars 71 forks source link

add sfast_extra_key to customize the cache key #85

Open huo-ju opened 9 months ago

huo-ju commented 9 months ago

Proposal:

add a sfast_extra_key to customize the cache key.

The current trace module key included hash_arg(args) and hash_arg(kwargs), but some args could be set up in other ways. (e.g. by attention_processor mixin).

Enabling extra kwargs can let users easily define their own trace key.