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