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,