Langchain-Chatchat/models/loader
Jingsong-Yan 421ce3da3a
Add device_map config to support chatglm2-6b (#734)
chatglm-6b和chatglm2-6b的参数命名不一致,本次提交旨在解决chatglm2-6b device_map 创建的问题。在chatglm_auto_configure_device_map 函数中新增了chatglm2-6b device_map 创建的相关代码。
2023-06-30 21:08:38 +08:00
..
__init__.py llm_model_dict 处理了loader的一些预设行为,如加载位置,模型名称,模型处理器实例, 定义checkpoint名称和远程路径 2023-05-18 22:54:41 +08:00
args.py temp 2023-06-12 12:57:41 +08:00
loader.py Add device_map config to support chatglm2-6b (#734) 2023-06-30 21:08:38 +08:00