Commit
·
0eed941
1
Parent(s):
feb19a1
Update h2oai_pipeline.py
Browse files- h2oai_pipeline.py +0 -1
h2oai_pipeline.py
CHANGED
|
@@ -36,4 +36,3 @@ class H2OTextGenerationPipeline(TextGenerationPipeline):
|
|
| 36 |
stopping_criteria = get_stopping(prompt_type, self.tokenizer, self.device, human=human, bot=bot)
|
| 37 |
generate_kwargs['stopping_criteria'] = stopping_criteria
|
| 38 |
return super()._forward(model_inputs, **generate_kwargs)
|
| 39 |
-
|
|
|
|
| 36 |
stopping_criteria = get_stopping(prompt_type, self.tokenizer, self.device, human=human, bot=bot)
|
| 37 |
generate_kwargs['stopping_criteria'] = stopping_criteria
|
| 38 |
return super()._forward(model_inputs, **generate_kwargs)
|
|
|