No, I do not want CUDA

2025-05-20 21:46:25 +03:00
parent c5495832c5
commit 0d7e2a4e60
1 changed files with 1 additions and 0 deletions
@@ -17,6 +17,7 @@ def apply(cls, text, **kwargs):


@mock.patch("prompt_control.utils.lora_name_to_file", find_file)
+@mock.patch("torch.cuda.current_device", lambda: "cpu")
 class GraphTests(unittest.TestCase):
    maxDiff = 4096