Make the "no GPU has been detected" message more descriptive

TheTerrasque · Mar 15, 2023 · 66256ac · 66256ac
1 parent 67d6247
commit 66256ac
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modules/models.py b/modules/models.py
@@ -98,7 +98,7 @@ def load_model(model_name):
         command = "AutoModelForCausalLM.from_pretrained"
         params = ["low_cpu_mem_usage=True"]
         if not shared.args.cpu and not torch.cuda.is_available():
-            print("Warning: no GPU has been detected.\nFalling back to CPU mode.\n")
+            print("Warning: torch.cuda.is_available() returned False.\nThis means that no GPU has been detected.\nFalling back to CPU mode.\n")
             shared.args.cpu = True
 
         if shared.args.cpu: