From 9c76d26b835c72d885033931b8e23cb0e6fb6d75 Mon Sep 17 00:00:00 2001 From: knc6 Date: Fri, 17 May 2024 01:47:37 -0400 Subject: [PATCH] No CPU test. --- atomgpt/tests/test_inverse.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/atomgpt/tests/test_inverse.py b/atomgpt/tests/test_inverse.py index 14fa9b7..40ac02c 100644 --- a/atomgpt/tests/test_inverse.py +++ b/atomgpt/tests/test_inverse.py @@ -1,4 +1,4 @@ -from atomgpt.inverse_models.mistral import FastMistralModel +# from atomgpt.inverse_models.mistral import FastMistralModel #def test_inverse_model(): # f = FastMistralModel.from_pretrained()