瀏覽代碼

fix mini bug

dbaranchuk 3 年之前
父節點
當前提交
a3e8c41389
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      src/client/remote_model.py

+ 1 - 1
src/client/remote_model.py

@@ -131,7 +131,7 @@ class DistributedBloomModel(BloomModel):
         hidden_states = self.word_embeddings_layernorm(inputs_embeds.float())
         output_shape = input_shape + (hidden_states.size(-1),)
 
-        if "ptune" in self.config.tuning_mode:
+        if self.config.tuning_mode and "ptune" in self.config.tuning_mode:
             hidden_states = self.h(hidden_states, prompts=intermediate_prompts)
         else:
             hidden_states = self.h(hidden_states)