mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2026-07-28 19:56:13 +08:00
[model] support MiniCPM-V-4.6 (#10472)
This commit is contained in:
@@ -205,6 +205,9 @@ class HuggingfaceEngine(BaseEngine):
|
||||
|
||||
gen_kwargs.pop("image_sizes", None)
|
||||
|
||||
if getattr(model.config, "model_type", None) == "minicpmv4_6":
|
||||
gen_kwargs["downsample_mode"] = os.getenv("DOWNSAMPLE_MODE", "16x")
|
||||
|
||||
return gen_kwargs, prompt_length
|
||||
|
||||
@staticmethod
|
||||
|
||||
Reference in New Issue
Block a user