Commit 15b3f9fb authored by dmMaze's avatar dmMaze
Browse files

allow float top p for chatgpts

parent 2b64bcbb

modules/translators/trans_chatgpt.py

+3 −3
Original line number Diff line number Diff line
@@ -69,7 +69,7 @@ class GPTTranslator(BaseTranslator): 
        'delay': 0.3,

        'max tokens': 4096,

        'temperature': 0.5,

        'top p': 1,

        'top p': 1.,

        # 'return prompt': False,

        'retry attempts': 5,

        'retry timeout': 15,
@@ -124,7 +124,7 @@ class GPTTranslator(BaseTranslator): 
        return self.params['max tokens']

    

    @property

    def top_p(self) -> int:

    def top_p(self) -> float:

        return self.params['top p']

    

    @property
@@ -167,7 +167,7 @@ class GPTTranslator(BaseTranslator): 
        else:

            return None



    def _assemble_prompts(self, queries: List[str], from_lang: str = None, to_lang: str = None, max_tokens = None) -> List[str]:

    def _assemble_prompts(self, queries: List[str], from_lang: str = None, to_lang: str = None, max_tokens = None):

        if from_lang is None:

            from_lang = self.lang_map[self.lang_source]

        if to_lang is None:

modules/translators/trans_chatgpt_exp.py

+3 −3
Original line number Diff line number Diff line
@@ -32,7 +32,7 @@ class GPTTranslatorExp(BaseTranslator): 
        'override model': '',

        'max tokens': 4096,

        'temperature': 0.5,

        'top p': 1,

        'top p': 1.,

        'retry attempts': 5,

        'retry timeout': 15,

        '3rd party api url': '',
@@ -84,7 +84,7 @@ class GPTTranslatorExp(BaseTranslator): 
        return self.params['max tokens']

    

    @property

    def top_p(self) -> int:

    def top_p(self) -> float:

        return self.params['top p']

    

    @property
@@ -95,7 +95,7 @@ class GPTTranslatorExp(BaseTranslator): 
    def retry_timeout(self) -> int:

        return self.params['retry timeout']



    def _assemble_prompts(self, queries: List[str], from_lang: str = None, to_lang: str = None, max_tokens = None) -> List[str]:

    def _assemble_prompts(self, queries: List[str], from_lang: str = None, to_lang: str = None, max_tokens = None):

        self.logger.debug("Assembling prompts...")

        if from_lang is None:

            from_lang = self.lang_map[self.lang_source]