google-ai-edge / ai-edge-torch

Supporting PyTorch models with the Google AI Edge TFLite runtime.
Apache License 2.0
378 stars 51 forks source link

In SD decoder/diffusion configuration, add a device type field to check if hlfb is needed. Also update the converter code to inspect device type flag. This hopefully will limit the number of places where `enable_hlfb` should be flipped. #327

Closed copybara-service[bot] closed 1 month ago

copybara-service[bot] commented 1 month ago

In SD decoder/diffusion configuration, add a device type field to check if hlfb is needed. Also update the converter code to inspect device type flag. This hopefully will limit the number of places where enable_hlfb should be flipped.