Upload folder using huggingface_hub
Browse files- amplify_te.py +2 -0
amplify_te.py
CHANGED
|
@@ -93,7 +93,9 @@ class AMPLIFYConfig(PretrainedConfig):
|
|
| 93 |
class AMPLIFYPreTrainedModel(PreTrainedModel):
|
| 94 |
"""AMPLIFY pre-trained model."""
|
| 95 |
|
|
|
|
| 96 |
config_class = AMPLIFYConfig
|
|
|
|
| 97 |
|
| 98 |
def _init_weights(self, module):
|
| 99 |
if isinstance(
|
|
|
|
| 93 |
class AMPLIFYPreTrainedModel(PreTrainedModel):
|
| 94 |
"""AMPLIFY pre-trained model."""
|
| 95 |
|
| 96 |
+
config: AMPLIFYConfig
|
| 97 |
config_class = AMPLIFYConfig
|
| 98 |
+
base_model_prefix = "amplify"
|
| 99 |
|
| 100 |
def _init_weights(self, module):
|
| 101 |
if isinstance(
|