Spaces:
Running
on
A10G
Running
on
A10G
Reuben Tan
commited on
Commit
•
2b6a2d1
1
Parent(s):
05aa141
test new path
Browse files
global_local/models/video_instruction_ft_model.py
CHANGED
@@ -125,7 +125,7 @@ class VideoInstructionFTLLAMA(Blip2Base):
|
|
125 |
logging.info('Loading Q-Former Done')
|
126 |
|
127 |
logging.info('Loading LLAMA Tokenizer')
|
128 |
-
self.llama_tokenizer = LlamaTokenizer.from_pretrained(llama_model, use_fast=False, token='
|
129 |
if self.llama_tokenizer.pad_token is None:
|
130 |
self.llama_tokenizer.pad_token = self.llama_tokenizer.unk_token
|
131 |
DEFAULT_IMAGE_PATCH_TOKEN = '<ImageHere>'
|
|
|
125 |
logging.info('Loading Q-Former Done')
|
126 |
|
127 |
logging.info('Loading LLAMA Tokenizer')
|
128 |
+
self.llama_tokenizer = LlamaTokenizer.from_pretrained(llama_model, use_fast=False, token=os.environ['LLAMA_TOKEN'])
|
129 |
if self.llama_tokenizer.pad_token is None:
|
130 |
self.llama_tokenizer.pad_token = self.llama_tokenizer.unk_token
|
131 |
DEFAULT_IMAGE_PATCH_TOKEN = '<ImageHere>'
|