tongxiaojun commited on
Commit
d7dbd92
·
1 Parent(s): f20bd07

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -2,11 +2,11 @@ import gradio as gr
2
  import random
3
  import time
4
 
5
- from transformers import BloomTokenizerFast, BloomForCausalLM
6
- path = 'redoop/opt-1.3b'
 
 
7
 
8
- tokenizer = BloomTokenizerFast.from_pretrained(path)
9
- model = BloomForCausalLM.from_pretrained(path)
10
  model.eval()
11
 
12
  #import intel_extension_for_pytorch as ipex
 
2
  import random
3
  import time
4
 
5
+ from transformers import AutoTokenizer, AutoModelForCausalLM
6
+
7
+ tokenizer = AutoTokenizer.from_pretrained("redoop/opt-1.3b")
8
+ model = AutoModelForCausalLM.from_pretrained("redoop/opt-1.3b")
9
 
 
 
10
  model.eval()
11
 
12
  #import intel_extension_for_pytorch as ipex