修复from_hf创建的模型在tfacc上运行不起来的问题

ztxz16 · May 9, 2024 · 6081a40 · 6081a40
1 parent a7dccd1
commit 6081a40
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/src/fastllm.cpp b/src/fastllm.cpp
@@ -1961,6 +1961,7 @@ namespace fastllm {
             realDims[1] *= 2;
         }
         this->weight[key] = Data(dataType, realDims);
+        this->weight[key].name = std::string(key);
         Data &data = this->weight[key];
         data.weightType = WeightType::LINEAR;
         data.UpdateUnitSize();
@@ -2007,6 +2008,7 @@ namespace fastllm {
     void WeightMap::AddWeight(const std::string &key, const std::vector<int> &dims, fastllm::DataType dataType,
                               fastllm::WeightType weightType, fastllm::DataType oriDataType, uint8_t *oriData, int groupCnt) {
         this->weight[key] = Data(dataType, dims);
+        this->weight[key].name = std::string(key);
         Data &data = this->weight[key];
         data.weightType = weightType;
         data.UpdateUnitSize();

diff --git a/third_party/tfacc/server b/third_party/tfacc/server