diff --git a/unimol/unimol/tasks/unimol.py b/unimol/unimol/tasks/unimol.py index f6b5242..ca787a8 100644 --- a/unimol/unimol/tasks/unimol.py +++ b/unimol/unimol/tasks/unimol.py @@ -235,3 +235,7 @@ def build_model(self, args): model = models.build_model(args, self) return model + + def disable_shuffling(self) -> bool: + return True +