From 4cc7b619f9e514a3540c4379cd432bb13c719932 Mon Sep 17 00:00:00 2001
From: Falko Habel <falko.habel@gmx.de>
Date: Tue, 11 Feb 2025 21:00:46 +0100
Subject: [PATCH] simplified comments

---
 example.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/example.py b/example.py
index ae3a35a..44e336e 100644
--- a/example.py
+++ b/example.py
@@ -3,7 +3,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 
 location = "Godala-moe"
 
-device = "cuda" # cpu when not using gpu
+device = "gpu" # or cpu
 tokenizer = AutoTokenizer.from_pretrained(location)
 
 model = AutoModelForCausalLM.from_pretrained(location).to(device)