公司服务器无法联网，有本地加载模型和数据集的例子嘛？

[mitergpt]

System Info

谢谢，公司服务无法联网

Reproduction

Put your message here.

Others

No response

[Ki-Seki]

examples/basic_modules/llm.py 里有使用本地模型的示例，修改 model_name_or_path 即可。

from memos.configs.llm import LLMConfigFactory
from memos.llms.factory import LLMFactory


config = LLMConfigFactory.model_validate(
    {
        "backend": "huggingface",
        "config": {
            "model_name_or_path": "Qwen/Qwen3-1.7B",
            "temperature": 0.8,
            "max_tokens": 1024,
            "top_p": 0.9,
            "top_k": 50,
        },
    }
)
llm = LLMFactory.from_config(config)
messages = [
    {"role": "user", "content": "Hello, who are you"},
]
response = llm.generate(messages)
print("Scenario 4:", response)
print("==" * 20)

[mitergpt]

谢谢 么么哒