Remove the Using GPU/CPU noisy message

radareorg · Sep 12, 2024 · 4460c98 · 4460c98
1 parent 97e7781
commit 4460c98
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/r2ai/models.py b/r2ai/models.py
@@ -143,9 +143,7 @@ def models():
 def gpulayers(ai):
     if "llm.gpu" in ai.env:
         if ai.env["llm.gpu"] == "true":
-            print("[r2ai] Using GPU")
             return -1
-    print("[r2ai] Using CPU")
     return 0
 
 def get_hf_llm(ai, repo_id, debug_mode, context_window):