Hugging Face's logo

Spaces:
agents-course
/
decoding_visualizer
Running on Zero

App Files Community
5
multimodalart HF Staff commited on
Commit
c05aa51
·
verified ·
1 Parent(s): b904851

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -6,6 +6,7 @@ import spaces
6
 
7
  tokenizer = AutoTokenizer.from_pretrained("HuggingFaceTB/SmolLM2-360M-Instruct")
8
  model = AutoModelForCausalLM.from_pretrained("HuggingFaceTB/SmolLM2-360M-Instruct")
 
9
 
10
  print("Loading finished.")
11
 
 
6
 
7
  tokenizer = AutoTokenizer.from_pretrained("HuggingFaceTB/SmolLM2-360M-Instruct")
8
  model = AutoModelForCausalLM.from_pretrained("HuggingFaceTB/SmolLM2-360M-Instruct")
9
+ model.to("cuda")
10
 
11
  print("Loading finished.")
12