Exclude D-FINE from using CUDA Graphs

This commit is contained in:
Nicolas Mowen 2025-12-16 09:32:50 -07:00
parent e7d047715d
commit 5ad6416764

View File

@ -170,6 +170,7 @@ class CudaGraphRunner(BaseModelRunner):
return model_type not in [
ModelTypeEnum.yolonas.value,
ModelTypeEnum.dfine.value,
EnrichmentModelTypeEnum.paddleocr.value,
EnrichmentModelTypeEnum.jina_v1.value,
EnrichmentModelTypeEnum.jina_v2.value,