Change compile backend to eager

qubvel-org · Jan 15, 2025 · a2b97d8 · a2b97d8
1 parent 524bcae
commit a2b97d8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/encoders/base.py b/tests/encoders/base.py
@@ -213,7 +213,7 @@ def test_compile(self):
         encoder = encoder.eval().to(default_device)
 
         torch.compiler.reset()
-        compiled_encoder = torch.compile(encoder, fullgraph=True, dynamic=True)
+        compiled_encoder = torch.compile(encoder, fullgraph=True, dynamic=True, backend="eager")
 
         if encoder._is_torch_compilable:
             compiled_encoder(sample)

diff --git a/tests/models/base.py b/tests/models/base.py
@@ -232,7 +232,7 @@ def test_compile(self):
         model = model.eval().to(default_device)
 
         torch.compiler.reset()
-        compiled_model = torch.compile(model, fullgraph=True, dynamic=True)
+        compiled_model = torch.compile(model, fullgraph=True, dynamic=True, backend="eager")
 
         with torch.inference_mode():
             compiled_model(sample)