Update transformers_plugin.py

transformers_plugin.py

@@ -9,9 +9,10 @@ class LlamaXFormersPlugin(TransformersPlugin):
  def process_model_pre_init(self, model):
  model_config = model.config
  replace_target_class(model, LlamaXFormersAttention, "LlamaAttention", init_kwargs={"config": model_config})
+ return model
  
 
  def process_model_post_init(self, model):
- pass
+ return model