artek0chumak commited on
Commit
542e6ca
1 Parent(s): 23f274a

reverse fix

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ import streamlit as st
11
  MODEL_NAME = "bigscience/test-bloomd-6b3" # select model you like
12
  # INITIAL_PEERS = ["/ip4/193.106.95.184/tcp/31000/p2p/QmSg7izCDtowVTACbUmWvEiQZNY4wgCQ9T9Doo66K59X6q"]
13
 
14
- tokenizer = transformers.BloomTokenizer.from_pretrained(MODEL_NAME)
15
  model = transformers.BloomForCausalLM.from_pretrained(MODEL_NAME)
16
 
17
  text = st.text_input('Enter some text')
 
11
  MODEL_NAME = "bigscience/test-bloomd-6b3" # select model you like
12
  # INITIAL_PEERS = ["/ip4/193.106.95.184/tcp/31000/p2p/QmSg7izCDtowVTACbUmWvEiQZNY4wgCQ9T9Doo66K59X6q"]
13
 
14
+ tokenizer = transformers.BloomTokenizerFast.from_pretrained(MODEL_NAME)
15
  model = transformers.BloomForCausalLM.from_pretrained(MODEL_NAME)
16
 
17
  text = st.text_input('Enter some text')