Patch fix - don't use safetensors for TF models (#30118)

* Patch fix - don't use safetensors for TF models

* Skip test for TF for now

* Update for another test
This commit is contained in:
amyeroberts
2024-04-08 13:29:20 +01:00
committed by GitHub
parent f5658732d5
commit 7f9aff910b
3 changed files with 6 additions and 5 deletions

View File

@@ -268,6 +268,7 @@ class TextGenerationPipelineTests(unittest.TestCase):
text_generator = TextGenerationPipeline(model=model, tokenizer=tokenizer)
return text_generator, ["This is a test", "Another test"]
@require_torch # See https://github.com/huggingface/transformers/issues/30117
def test_stop_sequence_stopping_criteria(self):
prompt = """Hello I believe in"""
text_generator = pipeline("text-generation", model="hf-internal-testing/tiny-random-gpt2")