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

develop #1

Merged
Fabel merged 5 commits from develop into main 2025-02-11 20:13:37 +00:00
Conversation 0 Commits 5 Files Changed 4 +1 -1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 4cc7b619f9 - Show all commits

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)