Spaces:
Sleeping
Sleeping
Reuben Tan
commited on
Commit
·
7001c0f
1
Parent(s):
24a4377
add tokens
Browse files
global_local/models/video_instruction_ft_model.py
CHANGED
|
@@ -149,6 +149,7 @@ class VideoInstructionFTLLAMA(Blip2Base):
|
|
| 149 |
self.llama_model = LlamaForCausalLM.from_pretrained(
|
| 150 |
llama_model,
|
| 151 |
torch_dtype=torch.bfloat16,
|
|
|
|
| 152 |
)
|
| 153 |
|
| 154 |
for name, param in self.llama_model.named_parameters():
|
|
|
|
| 149 |
self.llama_model = LlamaForCausalLM.from_pretrained(
|
| 150 |
llama_model,
|
| 151 |
torch_dtype=torch.bfloat16,
|
| 152 |
+
token=os.environ['LLAMA_TOKEN'],
|
| 153 |
)
|
| 154 |
|
| 155 |
for name, param in self.llama_model.named_parameters():
|