SwissArmyTransformer icon indicating copy to clipboard operation
SwissArmyTransformer copied to clipboard

Published 20 hours ago • verified publisher icon THUDM

Cannot use torch.compile with SAT

Open lijing1996 opened this issue 1 year ago • 0 comments

I tried to use torch.compile but failed with SAT. The reason is that self.transformer.hooks.clear() in base_model.py also clear the hooks of torch.compile?

lijing1996 avatar Oct 23 '23 07:10 lijing1996