minigpt42

Runtime error

Vision-CAIR commited on Apr 16, 2023

Commit

6947959

1 Parent(s): c93be5d

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')
-        self.llama_tokenizer = LlamaTokenizer.from_pretrained(llama_model, use_fast=False)
         self.llama_tokenizer.pad_token = self.llama_tokenizer.eos_token
         if llama_cache_dir:

         print('Loading Q-Former Done')
         print('Loading LLAMA')
+        self.llama_tokenizer = LlamaTokenizer.from_pretrained(llama_model, use_fast=False, cache_dir=llama_cache_dir)
         self.llama_tokenizer.pad_token = self.llama_tokenizer.eos_token
         if llama_cache_dir: