Spaces:
Running
on
Zero
Running
on
Zero
Update src/config.py
Browse files- 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
|
|
|
|
|
|
|
|
|
|
|
|
|
|