dbaranchuk 3 лет назад
Родитель
Сommit
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,