Vision-CAIR commited on
Commit
6aee8cd
β€’
1 Parent(s): 9367402

Update minigpt4/models/mini_gpt4.py

Browse files
Files changed (1) hide show
  1. minigpt4/models/mini_gpt4.py +1 -1
minigpt4/models/mini_gpt4.py CHANGED
@@ -75,7 +75,7 @@ class MiniGPT4(Blip2Base):
75
  for layer in self.Qformer.bert.encoder.layer:
76
  layer.output = None
77
  layer.intermediate = None
78
- self.load_from_pretrained(cache_dir=q_former_model)
79
 
80
  if freeze_qformer:
81
  for name, param in self.Qformer.named_parameters():
 
75
  for layer in self.Qformer.bert.encoder.layer:
76
  layer.output = None
77
  layer.intermediate = None
78
+ self.load_from_pretrained(url_or_filename=q_former_model)
79
 
80
  if freeze_qformer:
81
  for name, param in self.Qformer.named_parameters():