Spaces:

Vision-CAIR
/

minigpt4

Running on A10G

Vision-CAIR commited on Jul 3, 2023

Commit

d66bcec

•

1 Parent(s): 10b2592

Update minigpt4/models/mini_gpt4.py

Files changed (1) hide show

minigpt4/models/mini_gpt4.py CHANGED Viewed

@@ -87,7 +87,7 @@ class MiniGPT4(Blip2Base):
         print('Loading Q-Former Done')
         print('Loading LLAMA')
-        print('API_token': os.environ["API_TOKEN"])
         self.llama_tokenizer = LlamaTokenizer.from_pretrained('Vision-CAIR/vicuna-7b', use_fast=False, use_auth_token=os.environ["API_TOKEN"])
         self.llama_tokenizer.pad_token = self.llama_tokenizer.eos_token

         print('Loading Q-Former Done')
         print('Loading LLAMA')
+        print('API_token:', os.environ["API_TOKEN"])
         self.llama_tokenizer = LlamaTokenizer.from_pretrained('Vision-CAIR/vicuna-7b', use_fast=False, use_auth_token=os.environ["API_TOKEN"])
         self.llama_tokenizer.pad_token = self.llama_tokenizer.eos_token