diff --git a/video_miniGPT4/minigpt4/models/eva_vit.py b/video_miniGPT4/minigpt4/models/eva_vit.py index da2a9c6..13ce390 100644 --- a/video_miniGPT4/minigpt4/models/eva_vit.py +++ b/video_miniGPT4/minigpt4/models/eva_vit.py @@ -56,7 +56,7 @@ def forward(self, x): x = self.fc1(x) x = self.act(x) # x = self.drop(x) - # commit this for the orignal BERT implement + # commit this for the original BERT implement x = self.fc2(x) x = self.drop(x) return x @@ -442,4 +442,4 @@ def create_eva_vit_g( if precision == "fp16": # model.to("cuda") convert_weights_to_fp16(model) - return model \ No newline at end of file + return model