sitammeur commited on
Commit
93437e6
1 Parent(s): cdcc84d

Update src/config.py

Browse files
Files changed (1) hide show
  1. src/config.py +12 -19
src/config.py CHANGED
@@ -1,19 +1,12 @@
1
- # Model settings
2
- device = "cuda"
3
- model_name = "openbmb/MiniCPM-V-2_6"
4
-
5
- # Decoding settings
6
- sampling = True
7
- stream = True
8
- top_p = 0.8
9
- top_k = 100
10
- temperature = 0.7
11
- repetition_penalty = 1.05
12
- max_new_tokens = 2048
13
-
14
- # System Prompt
15
- system_prompt = """
16
- You are an AI assistant specialized in visual content analysis.
17
- Given a video and a related question, analyze the video thoroughly and provide a precise and informative answer based on the visible content.
18
- Ensure your response is clear, accurate, and directly addresses the question.
19
- """
 
1
+ # Model settings
2
+ device = "cuda"
3
+ model_name = "openbmb/MiniCPM-V-2_6"
4
+
5
+ # Decoding settings
6
+ sampling = True
7
+ stream = True
8
+ top_p = 0.8
9
+ top_k = 100
10
+ temperature = 0.7
11
+ repetition_penalty = 1.05
12
+ max_new_tokens = 2048