Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
4049b6f
1
Parent(s):
14a3ecc
[Enhancement] Update to VideoLLaMA2.1 (#3)
Browse files- Update app.py (e1c10d2c2ae20449c4ba137b6ef238bfc71007de)
Co-authored-by: Kehan Li <lkhl@users.noreply.huggingface.co>
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 |
|