Machine-Learning/Godala
Machine-Learning
/
Godala
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 1 Wiki Activity

simplified comments

This commit is contained in:
Falko Victor Habel 2025-02-11 21:00:46 +01:00
parent 3bcf8a25b2
commit 4cc7b619f9
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
example.py
View File

@ -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)