dwb2023 commited on
Commit
90a9957
·
verified ·
1 Parent(s): cbf8993

update transformers import

Browse files
Files changed (1) hide show
  1. app_dialogue.py +2 -4
app_dialogue.py CHANGED
@@ -1,5 +1,5 @@
1
  import os
2
- import subprocess
3
 
4
  # Install flash attention
5
  # subprocess.run(
@@ -22,9 +22,7 @@ import io
22
  import datasets
23
 
24
  import gradio as gr
25
- from transformers import AutoImageProcessor, AutoTokenizer, TextIteratorStreamer, StoppingCriteria
26
- from transformers import AutoModelForVision2Seq
27
-
28
 
29
  # model_name_or_path = "Salesforce/xgen-mm-phi3-mini-instruct-interleave-r-v1.5"
30
  # model = AutoModelForVision2Seq.from_pretrained(model_name_or_path, trust_remote_code=True)
 
1
  import os
2
+ # import subprocess
3
 
4
  # Install flash attention
5
  # subprocess.run(
 
22
  import datasets
23
 
24
  import gradio as gr
25
+ from transformers import AutoModelForVision2Seq, AutoTokenizer, AutoImageProcessor, LogitsProcessor
 
 
26
 
27
  # model_name_or_path = "Salesforce/xgen-mm-phi3-mini-instruct-interleave-r-v1.5"
28
  # model = AutoModelForVision2Seq.from_pretrained(model_name_or_path, trust_remote_code=True)