Didier commited on
Commit
39eaec8
·
verified ·
1 Parent(s): 5a1631f

Update vlm.py

Browse files
Files changed (1) hide show
  1. vlm.py +6 -0
vlm.py CHANGED
@@ -7,7 +7,13 @@ Date: 2025-05-08
7
 
8
  from transformers import AutoProcessor
9
  from transformers import Mistral3ForConditionalGeneration
 
 
 
 
10
  import torch
 
 
11
 
12
  #
13
  # Load the model: OPEA/Mistral-Small-3.1-24B-Instruct-2503-int4-AutoRound-awq-sym
 
7
 
8
  from transformers import AutoProcessor
9
  from transformers import Mistral3ForConditionalGeneration
10
+ from transformers import TextIteratorStreamer
11
+ from threading import Thread
12
+ import re
13
+ import time
14
  import torch
15
+ import spaces
16
+
17
 
18
  #
19
  # Load the model: OPEA/Mistral-Small-3.1-24B-Instruct-2503-int4-AutoRound-awq-sym