ToletiSri commited on
Commit
d64312a
·
verified ·
1 Parent(s): 1cf8caa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -22,8 +22,12 @@ class _MLPVectorProjector(nn.Module):
22
 
23
  model_name = "microsoft/phi-2"
24
 
25
- phi2_text = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype="auto", trust_remote_code=True)
26
- tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
 
 
 
 
27
 
28
 
29
  def textMode(text, count):
 
22
 
23
  model_name = "microsoft/phi-2"
24
 
25
+ import transformers
26
+ print(----------version--------------)
27
+ print(transformers.__version__)
28
+
29
+ #phi2_text = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype="auto", trust_remote_code=True)
30
+ #tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
31
 
32
 
33
  def textMode(text, count):