Update transformers_plugin.py
Browse files- transformers_plugin.py +6 -0
transformers_plugin.py
CHANGED
|
@@ -4,4 +4,10 @@ class LlamaXFormersPlugin(TransformersPlugin):
|
|
| 4 |
def __init__(self, config):
|
| 5 |
pass
|
| 6 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 7 |
|
|
|
|
| 4 |
def __init__(self, config):
|
| 5 |
pass
|
| 6 |
|
| 7 |
+
def process_model_pre_init(self, model):
|
| 8 |
+
pass
|
| 9 |
+
|
| 10 |
+
def process_model_post_init(self, model):
|
| 11 |
+
pass
|
| 12 |
+
|
| 13 |
|