Spaces:

rxtan
/

Koala-video-llm

Sleeping

Reuben Tan commited on Mar 13

Commit

7001c0f

•

1 Parent(s): 24a4377

add tokens

Files changed (1) hide show

global_local/models/video_instruction_ft_model.py CHANGED Viewed

@@ -149,6 +149,7 @@ class VideoInstructionFTLLAMA(Blip2Base):
             self.llama_model = LlamaForCausalLM.from_pretrained(
                 llama_model,
                 torch_dtype=torch.bfloat16,
             )
         for name, param in self.llama_model.named_parameters():

             self.llama_model = LlamaForCausalLM.from_pretrained(
                 llama_model,
                 torch_dtype=torch.bfloat16,
+                token=os.environ['LLAMA_TOKEN'],
             )
         for name, param in self.llama_model.named_parameters():