acbdkk commited on
Commit
24185c8
1 Parent(s): c61b7c5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,9 +1,9 @@
1
  import streamlit as st
2
  import transformers
3
- from transformers import LlamaTokenizer
4
  import torch
5
  model = "PY007/TinyLlama-1.1B-Chat-v0.1"
6
- tokenizer = transformers.LlamaTokenizer.from_pretrained(model)
7
  pipeline = transformers.pipeline("text-generation",model=model,torch_dtype=torch.bfloat16)
8
 
9
  prompt = "What is 653+2343?"
 
1
  import streamlit as st
2
  import transformers
3
+ from transformers import LLaMATokenizer
4
  import torch
5
  model = "PY007/TinyLlama-1.1B-Chat-v0.1"
6
+ tokenizer = transformers.LLaMATokenizer.from_pretrained(model)
7
  pipeline = transformers.pipeline("text-generation",model=model,torch_dtype=torch.bfloat16)
8
 
9
  prompt = "What is 653+2343?"