Commit 5c2b94c8 authored by LysandreJik's avatar LysandreJik
Browse files

Changed string so that Circle CI accepts the warning

parent 87747518
......@@ -381,8 +381,9 @@ class PreTrainedModel(nn.Module):
self.config.pruned_heads[int(layer)].append(head)
to_be_pruned[int(layer)].append(head)
else:
logger.warning(f"Tried to remove head {head} of layer {layer} but it was already removed. "
f"The removed heads are {heads_to_prune}")
logger.warning("Tried to remove head " + head +
" of layer " + layer +
" but it was already removed. The current removed heads are " + heads_to_prune)
base_model._prune_heads(to_be_pruned)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment