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

Add test to check that the external kv cache op is present in the model. #280

Closed copybara-service[bot] closed 1 month ago

copybara-service[bot] commented 1 month ago

Add test to check that the external kv cache op is present in the model.