Tonic commited on
Commit
2df0bb2
Β·
1 Parent(s): c93fb24

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ import os
8
  device = "cuda" if torch.cuda.is_available() else "cpu"
9
 
10
  # Load tokenizer and model
11
- tokenizer = AutoTokenizer.from_pretrained('stabilityai/stablelm-zephyr-3b', token=hf_token)
12
  model = AutoModelForCausalLM.from_pretrained(
13
  'stabilityai/stablelm-zephyr-3b',
14
  trust_remote_code=True,
 
8
  device = "cuda" if torch.cuda.is_available() else "cpu"
9
 
10
  # Load tokenizer and model
11
+ tokenizer = AutoTokenizer.from_pretrained('stabilityai/stablelm-zephyr-3b')
12
  model = AutoModelForCausalLM.from_pretrained(
13
  'stabilityai/stablelm-zephyr-3b',
14
  trust_remote_code=True,