working on automodels
This commit is contained in:
1 parent
58830807d1
commit
b90e29d52c
6 files changed
+289
-14
No files matched your search
@@ -171,7 +171,7 @@ class OpenAIGPTConfig(PretrainedConfig):
|
||||
predict_special_tokens=True,
|
||||
|
||||
num_labels=1,
|
||||
summary_type='token_ids',
|
||||
summary_type='cls_index',
|
||||
summary_use_proj=True,
|
||||
summary_activation=None,
|
||||
summary_proj_to_labels=True,
|
||||
|
||||
Reference in new issue
Block a user