dbaranchuk пре 3 година
родитељ
комит
f66547bac5
1 измењених фајлова са 0 додато и 1 уклоњено
  1. 0 1
      src/bloom/block.py

+ 0 - 1
src/bloom/block.py

@@ -202,7 +202,6 @@ class BloomBlock(nn.Module):
     def forward(
         self,
         hidden_states,
-        prompts=None,
         layer_past=None,
         attention_mask=None,
         head_mask=None,