[Bugfix][CI]Fix HSDP regional compile boundary (#4668)

Signed-off-by: bjf-frz <frz123db@gmail.com>
This commit is contained in:
bjf-frz authored and GitHub committed 2026-06-25 02:52:52 +00:00
1 parent f6e8b77642
commit d612233bef
6 files changed
+176 -6

No files matched your search

@@ -88,6 +88,7 @@ class DiffusionModelRunner(OmniConnectorModelRunnerMixin):
model = getattr(self.pipeline, attr_name, None)
if model is None:
return
try:
setattr(self.pipeline, attr_name, regionally_compile(model, dynamic=True))
logger.info("Model runner: %s compiled with torch.compile.", attr_name)