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,