Spaces:

Vision-CAIR
/

minigpt4

Running on A10G

Vision-CAIR commited on Apr 16, 2023

Commit

6aee8cd

•

1 Parent(s): 9367402

Update minigpt4/models/mini_gpt4.py

Files changed (1) hide show

minigpt4/models/mini_gpt4.py CHANGED Viewed

@@ -75,7 +75,7 @@ class MiniGPT4(Blip2Base):
         for layer in self.Qformer.bert.encoder.layer:
             layer.output = None
             layer.intermediate = None
-        self.load_from_pretrained(cache_dir=q_former_model)
         if freeze_qformer:
             for name, param in self.Qformer.named_parameters():

         for layer in self.Qformer.bert.encoder.layer:
             layer.output = None
             layer.intermediate = None
+        self.load_from_pretrained(url_or_filename=q_former_model)
         if freeze_qformer:
             for name, param in self.Qformer.named_parameters():