瀏覽代碼

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

oobabooga 2 年之前
父節點
當前提交
66256ac1dd
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      modules/models.py

+ 1 - 1
modules/models.py

@@ -98,7 +98,7 @@ def load_model(model_name):
         command = "AutoModelForCausalLM.from_pretrained"
         command = "AutoModelForCausalLM.from_pretrained"
         params = ["low_cpu_mem_usage=True"]
         params = ["low_cpu_mem_usage=True"]
         if not shared.args.cpu and not torch.cuda.is_available():
         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
             shared.args.cpu = True
 
 
         if shared.args.cpu:
         if shared.args.cpu: