Spaces:
Running
Running
lym0302
commited on
Commit
·
9ebfdc2
1
Parent(s):
28e0d96
device
Browse files
third_party/VideoLLaMA2/videollama2/model/__init__.py
CHANGED
|
@@ -53,7 +53,7 @@ VLLMConfigs = {
|
|
| 53 |
|
| 54 |
|
| 55 |
def load_pretrained_model(model_path, model_base, model_name, load_8bit=False, load_4bit=False, device_map="auto", device="cuda", use_flash_attn=False, **kwargs):
|
| 56 |
-
|
| 57 |
if 'token' in kwargs:
|
| 58 |
token = kwargs['token']
|
| 59 |
else:
|
|
|
|
| 53 |
|
| 54 |
|
| 55 |
def load_pretrained_model(model_path, model_base, model_name, load_8bit=False, load_4bit=False, device_map="auto", device="cuda", use_flash_attn=False, **kwargs):
|
| 56 |
+
print("00000000000000000000000000: ", device)
|
| 57 |
if 'token' in kwargs:
|
| 58 |
token = kwargs['token']
|
| 59 |
else:
|