Browse Source
Remove useless parameter
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
1 changed files with
0 additions and
1 deletions
-
codebert.py
|
|
@ -45,7 +45,6 @@ class CodeBert(NNOperator): |
|
|
|
def __init__(self, model_name: str = 'huggingface/CodeBERTa-small-v1', device: str = None): |
|
|
|
super().__init__() |
|
|
|
self.model_name = model_name |
|
|
|
assert modality in ['nlp', 'code'], 'Invalid modality value. Accept only "nlp" or "code".' |
|
|
|
|
|
|
|
if device is None: |
|
|
|
device = 'cuda' if torch.cuda.is_available() else 'cpu' |
|
|
|