ivanwu/jarvis-models
1
0
Fork 0
mirror of https://github.com/BoardWare-Genius/jarvis-models.git synced 2025-12-13 16:53:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
0Xiao0
2025-03-19 16:43:24 +08:00
parent 7e425ea574
commit a8e2923f9c
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/blackbox/vlms.py
View File

@ -76,6 +76,8 @@ class VLMS(Blackbox):
"top_k": self.top_k, "top_k": self.top_k,
"ignore_eos": self.ignore_eos, "ignore_eos": self.ignore_eos,
"skip_special_tokens": self.skip_special_tokens, "skip_special_tokens": self.skip_special_tokens,
# "system_prompt":"",
# "vlm_model_name":" ",
} }
@ -99,7 +101,7 @@ class VLMS(Blackbox):
history: a list history: a list
""" """
if settings: if settings:
for k in settings: for k in list(settings.keys()):
if k not in self.settings: 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())) print("Warning: '{}' is not a support argument and ignore this argment, check the arguments {}".format(k,self.settings.keys()))
settings.pop(k) settings.pop(k)