Upload folder using huggingface_hub
Browse files- README.md +0 -1
- amplify_te.py +2 -0
README.md
CHANGED
|
@@ -1,5 +1,4 @@
|
|
| 1 |
---
|
| 2 |
-
library_name: transformers
|
| 3 |
license: mit
|
| 4 |
datasets:
|
| 5 |
- chandar-lab/UR100P
|
|
|
|
| 1 |
---
|
|
|
|
| 2 |
license: mit
|
| 3 |
datasets:
|
| 4 |
- chandar-lab/UR100P
|
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(
|