huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-12 04:44:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Removing the dead code.

Browse Source
This commit is contained in:
Nicolas Patry 2024-07-03 13:34:37 +00:00
parent 9cc58d1cb3
commit fbf38c997c
No known key found for this signature in database
GPG Key ID: E939E8CC91A1C674
2 changed files with 0 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
server/text_generation_server/models/causal_lm.py
View File

@ -644,12 +644,6 @@ class CausalLM(Model):
def batch_type(self) -> Type[CausalLMBatch]:
return self.batch
# This is not used anymore
# def decode(self, generated_ids: List[int]) -> str:
# return self.tokenizer.decode(
# generated_ids, skip_special_tokens=True, clean_up_tokenization_spaces=False
# )
def forward(
self, input_ids, attention_mask, position_ids, past_key_values: Optional = None
) -> Tuple[

6
server/text_generation_server/models/seq2seq_lm.py
View File

@ -677,12 +677,6 @@ class Seq2SeqLM(Model):
def batch_type(self) -> Type[Seq2SeqLMBatch]:
return Seq2SeqLMBatch
# Not used anymore
# def decode(self, decoder_ids: List[int]) -> str:
# return self.tokenizer.decode(
# decoder_ids, skip_special_tokens=True, clean_up_tokenization_spaces=False
# )
def forward(
self,
input_ids,