remove config class in model
Browse files- modeling_gptpangu.py +2 -2
modeling_gptpangu.py
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
"""PyTorch PanguAlpha GPT2 Model"""
|
| 2 |
-
from .configuration_gptpangu import GPTPanguConfig
|
| 3 |
|
| 4 |
from typing import Tuple
|
| 5 |
import math
|
|
@@ -207,7 +207,7 @@ class GPTPanguPreTrainedModel(PreTrainedModel):
|
|
| 207 |
models.
|
| 208 |
"""
|
| 209 |
|
| 210 |
-
config_class = GPTPanguConfig
|
| 211 |
base_model_prefix = "transformer"
|
| 212 |
supports_gradient_checkpointing = True
|
| 213 |
|
|
|
|
| 1 |
"""PyTorch PanguAlpha GPT2 Model"""
|
| 2 |
+
# from .configuration_gptpangu import GPTPanguConfig
|
| 3 |
|
| 4 |
from typing import Tuple
|
| 5 |
import math
|
|
|
|
| 207 |
models.
|
| 208 |
"""
|
| 209 |
|
| 210 |
+
# config_class = GPTPanguConfig
|
| 211 |
base_model_prefix = "transformer"
|
| 212 |
supports_gradient_checkpointing = True
|
| 213 |
|