diff --git a/src/blackbox/vlms.py b/src/blackbox/vlms.py index 1a65fbf..2ff2b15 100644 --- a/src/blackbox/vlms.py +++ b/src/blackbox/vlms.py @@ -76,6 +76,8 @@ class VLMS(Blackbox): "top_k": self.top_k, "ignore_eos": self.ignore_eos, "skip_special_tokens": self.skip_special_tokens, + # "system_prompt":"", + # "vlm_model_name":" ", } @@ -99,7 +101,7 @@ class VLMS(Blackbox): history: a list """ if settings: - for k in settings: + for k in list(settings.keys()): if k not in self.settings: print("Warning: '{}' is not a support argument and ignore this argment, check the arguments {}".format(k,self.settings.keys())) settings.pop(k)