Upload modeling_cocom.py
Browse files- modeling_cocom.py +1 -1
modeling_cocom.py
CHANGED
|
@@ -61,7 +61,7 @@ class COCOMConfig(PretrainedConfig):
|
|
| 61 |
|
| 62 |
model_type = "COCOM"
|
| 63 |
def __init__(self,
|
| 64 |
-
decoder_model_name="google-t5/t5-
|
| 65 |
quantization = 'no',
|
| 66 |
generation_top_k = 1,
|
| 67 |
sep = False,
|
|
|
|
| 61 |
|
| 62 |
model_type = "COCOM"
|
| 63 |
def __init__(self,
|
| 64 |
+
decoder_model_name="google-t5/t5-base",
|
| 65 |
quantization = 'no',
|
| 66 |
generation_top_k = 1,
|
| 67 |
sep = False,
|