diff --git a/mediapipe/tasks/cc/vision/image_generator/diffuser/diffuser_gpu.h b/mediapipe/tasks/cc/vision/image_generator/diffuser/diffuser_gpu.h index 85738b80b..613c8e0f2 100644 --- a/mediapipe/tasks/cc/vision/image_generator/diffuser/diffuser_gpu.h +++ b/mediapipe/tasks/cc/vision/image_generator/diffuser/diffuser_gpu.h @@ -32,6 +32,7 @@ enum DiffuserModelType { kDiffuserModelTypeDistilledGldm, kDiffuserModelTypeSd2Base, kDiffuserModelTypeTigo, + kDiffuserModelTypeTigoUfo, }; enum DiffuserPriorityHint {