Spaces:
Running
on
Zero
Running
on
Zero
[Enhancement] Update to VideoLLaMA2.1 (#3)
Browse files- Update app.py (e1c10d2c2ae20449c4ba137b6ef238bfc71007de)
Co-authored-by: Kehan Li <[email protected]>
app.py
CHANGED
|
@@ -180,7 +180,7 @@ def clear_history(message, chatbot):
|
|
| 180 |
# 2. The operation or tensor which requires cuda are limited in those functions wrapped via spaces.GPU
|
| 181 |
# 3. The function can't return tensor or other cuda objects.
|
| 182 |
|
| 183 |
-
model_path = 'DAMO-NLP-SG/VideoLLaMA2-7B-16F'
|
| 184 |
|
| 185 |
handler = Chat(model_path, load_8bit=False, load_4bit=True)
|
| 186 |
|
|
|
|
| 180 |
# 2. The operation or tensor which requires cuda are limited in those functions wrapped via spaces.GPU
|
| 181 |
# 3. The function can't return tensor or other cuda objects.
|
| 182 |
|
| 183 |
+
model_path = 'DAMO-NLP-SG/VideoLLaMA2.1-7B-16F'
|
| 184 |
|
| 185 |
handler = Chat(model_path, load_8bit=False, load_4bit=True)
|
| 186 |
|