fix

huggingface · Feb 13, 2024 · 295c190 · 295c190
1 parent f7e4069
commit 295c190
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/optimum_benchmark/backends/pytorch/backend.py b/optimum_benchmark/backends/pytorch/backend.py
@@ -79,7 +79,7 @@ def __init__(self, config: PyTorchConfig):
             self.pretrained_model.generation_config.cache_implementation = self.config.cache_implementation
 
         # Eval mode
-        if self.config.eval_mode and self.config.library == "diffusers":
+        if self.config.eval_mode and self.config.library != "diffusers":
             LOGGER.info("\t+ Turning on model's eval mode")
             self.pretrained_model.eval()