add eval method

chaiNNer-org · Nov 18, 2023 · 9388340 · 9388340
1 parent c6c72e1
commit 9388340
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/spandrel/__helpers/model_descriptor.py b/src/spandrel/__helpers/model_descriptor.py
@@ -149,6 +149,10 @@ def cpu(self):
         self.model.cpu()
         return self
 
+    def eval(self):
+        self.model.eval()
+        return self
+
     def __call__(self, *args, **kwargs):  # noqa: ANN002
         return self.model(*args, **kwargs)