Skip to content

Commit da3d0b6

Browse files
Update phishing_email_detection_gpt2.py
More debugging
1 parent db47c42 commit da3d0b6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

phishing_email_detection_gpt2.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -694,7 +694,7 @@ def get_config(self):
694694

695695
@classmethod
696696
def from_config(cls, config):
697-
config_obj = CerebrosAutoregressiveTextGeneratorConfig.from_config(config['config'])
697+
config_obj = CerebrosNotGPTConfig.from_config(config['config'])
698698
return cls(config=config_obj)
699699

700700
def generate(self, token_ids, do_sample=False, max_new_tokens=None):

0 commit comments

Comments
 (0)