Hugging Face's logo

Spaces:
Kkordik
/
GPU-SmolLM-Instruct
Runtime error

App Files Community
Kkordik commited on
Commit
35200a2
·
verified ·
1 Parent(s): 93893a2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ h3 {
33
  # pip install transformers
34
  from transformers import AutoModelForCausalLM, AutoTokenizer
35
 
36
- device = "cpu" # for GPU usage or "cpu" for CPU usage
37
 
38
  tokenizer0 = AutoTokenizer.from_pretrained(MODEL_LIST[0])
39
  model0 = AutoModelForCausalLM.from_pretrained(MODEL_LIST[0]).to(device)
 
33
  # pip install transformers
34
  from transformers import AutoModelForCausalLM, AutoTokenizer
35
 
36
+ device = "cuda"
37
 
38
  tokenizer0 = AutoTokenizer.from_pretrained(MODEL_LIST[0])
39
  model0 = AutoModelForCausalLM.from_pretrained(MODEL_LIST[0]).to(device)