This repository was archived by the owner on Jul 22, 2025. It is now read-only.
  
  
  
  
          Support for BertModel, BartModel (AutoModel)
          #29
        
      
  Add this suggestion to a batch that can be applied as a single commit.
  This suggestion is invalid because no changes were made to the code.
  Suggestions cannot be applied while the pull request is closed.
  Suggestions cannot be applied while viewing a subset of changes.
  Only one suggestion per line can be applied in a batch.
  Add this suggestion to a batch that can be applied as a single commit.
  Applying suggestions on deleted lines is not supported.
  You must change the existing code in this line in order to create a valid suggestion.
  Outdated suggestions cannot be applied.
  This suggestion has been applied or marked resolved.
  Suggestions cannot be applied from pending reviews.
  Suggestions cannot be applied on multi-line comments.
  Suggestions cannot be applied while the pull request is queued to merge.
  Suggestion cannot be applied right now. Please check back later.
  
    
  
    
Until now: the library supports pruning for
BertFor*(AutoModelFor*).Now: you can load both
BertFor*andBertModel(or just useAutoModel).What's new: I upgraded the pattern prefix regexp so it optionally captures
bert.(ormodel.for Bart).BertFor*start withbert.encoder.layer..BertModelstart with justencoder.layer..