Sfoglia il codice sorgente

rm redundant kwarg

dbaranchuk 3 anni fa
parent
commit
f66547bac5
1 ha cambiato i file con 0 aggiunte e 1 eliminazioni
  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,