3 years ago · f66547bac5
--- a/src/bloom/block.py
+++ b/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,