Update vitGPT.py

vitGPT.py

@@ -278,7 +278,7 @@ class VisionGPT2Model(nn.Module):
         lm_logits = self.lm_head(input_ids[:,[-1],:])
         return lm_logits
 
-    def generate(self,image,sequence,tokenizer,max_tokens=50,temperature=1.0,deterministic=False):
+    def generate(self,image,sequence,tokenizer,max_tokens,temperature,deterministic=False):
         for _ in range(max_tokens):
             out = self(image,sequence)
             out = out[:,-1,:] / temperature
@@ -337,8 +337,8 @@ def generate_caption(image,max_tokens,temperature,deterministic=True):
         image,
         sequence,
         tokenizer,
-        max_tokens=max_tokens,
-        temperature=temperature,
+        max_tokens,
+        temperature,
         deterministic=deterministic, 
         
     )