Hugging Face's logo

Spaces:
nroggendorff
/
smallama
Paused

App Files Community
1
nroggendorff commited on
Commit
79de183
·
verified ·
1 Parent(s): 2af0ecd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, TextIteratorStream
4
  from threading import Thread
5
  import torch
6
 
7
- MODEL_ID = "nroggendorff/smallama-it"
8
 
9
  tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
10
  model = AutoModelForCausalLM.from_pretrained(
 
4
  from threading import Thread
5
  import torch
6
 
7
+ MODEL_ID = "nroggendorff/smallama-7b-it"
8
 
9
  tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
10
  model = AutoModelForCausalLM.from_pretrained(