Skip to content
项目
群组
代码片段
帮助
当前项目
正在载入...
登录 / 注册
切换导航面板
J
jinchat-server
概览
概览
详情
活动
周期分析
版本库
存储库
文件
提交
分支
标签
贡献者
分支图
比较
统计图
问题
0
议题
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
CI / CD
CI / CD
流水线
作业
日程表
图表
维基
Wiki
代码片段
代码片段
成员
成员
折叠边栏
关闭边栏
活动
图像
聊天
创建新问题
作业
提交
问题看板
Open sidebar
aigc-pioneer
jinchat-server
Commits
db0fcf4e
提交
db0fcf4e
authored
7月 14, 2023
作者:
hzg0601
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
print error
上级
760abab1
显示空白字符变更
内嵌
并排
正在显示
1 个修改的文件
包含
2 行增加
和
0 行删除
+2
-0
loader.py
models/loader/loader.py
+2
-0
没有找到文件。
models/loader/loader.py
浏览文件 @
db0fcf4e
...
@@ -454,6 +454,7 @@ class LoaderCheckPoint:
...
@@ -454,6 +454,7 @@ class LoaderCheckPoint:
self
.
model_config
.
pre_seq_len
=
prefix_encoder_config
[
'pre_seq_len'
]
self
.
model_config
.
pre_seq_len
=
prefix_encoder_config
[
'pre_seq_len'
]
self
.
model_config
.
prefix_projection
=
prefix_encoder_config
[
'prefix_projection'
]
self
.
model_config
.
prefix_projection
=
prefix_encoder_config
[
'prefix_projection'
]
except
Exception
as
e
:
except
Exception
as
e
:
print
(
e
)
print
(
"加载PrefixEncoder config.json失败"
)
print
(
"加载PrefixEncoder config.json失败"
)
self
.
model
,
self
.
tokenizer
=
self
.
_load_model
()
self
.
model
,
self
.
tokenizer
=
self
.
_load_model
()
...
@@ -471,6 +472,7 @@ class LoaderCheckPoint:
...
@@ -471,6 +472,7 @@ class LoaderCheckPoint:
self
.
model
.
transformer
.
prefix_encoder
.
load_state_dict
(
new_prefix_state_dict
)
self
.
model
.
transformer
.
prefix_encoder
.
load_state_dict
(
new_prefix_state_dict
)
self
.
model
.
transformer
.
prefix_encoder
.
float
()
self
.
model
.
transformer
.
prefix_encoder
.
float
()
except
Exception
as
e
:
except
Exception
as
e
:
print
(
e
)
print
(
"加载PrefixEncoder模型参数失败"
)
print
(
"加载PrefixEncoder模型参数失败"
)
# llama-cpp模型(至少vicuna-13b)的eval方法就是自身,其没有eval方法
# llama-cpp模型(至少vicuna-13b)的eval方法就是自身,其没有eval方法
if
not
self
.
is_llamacpp
:
if
not
self
.
is_llamacpp
:
...
...
编写
预览
Markdown
格式
0%
重试
或
添加新文件
添加附件
取消
您添加了
0
人
到此讨论。请谨慎行事。
请先完成此评论的编辑!
取消
请
注册
或者
登录
后发表评论