diff --git a/core/compiler.cpp b/core/compiler.cpp
index 7b58dbb2c1..898f2f1295 100644
--- a/core/compiler.cpp
+++ b/core/compiler.cpp
@@ -426,7 +426,7 @@ torch::jit::Module CompileGraph(const torch::jit::Module& mod, CompileSpec cfg)
       auto outputIsCollection = conversion::OutputIsCollection(g->block());
       if (cfg.partition_info.enabled &&
           (cfg.lower_info.forced_fallback_modules.size() == 0 &&
-           cfg.partition_info.forced_fallback_operators.size() == 0 && isBlockConvertible)) {
+           cfg.partition_info.forced_fallback_operators.size() == 0 && isBlockConvertible) && !outputIsCollection) {
         LOG_INFO("Skipping partitioning since model is fully supported");
       }