Revert workaround for TF safetensors loading (#30128)

* See if we can get tests to pass with the fixed weights

* See if we can get tests to pass with the fixed weights

* Replace the revisions now that we don't need them anymore
This commit is contained in:
Matt
2024-04-09 11:04:18 +01:00
committed by GitHub
parent 841e87ef4f
commit ec59a42192
3 changed files with 5 additions and 6 deletions

View File

@@ -268,7 +268,6 @@ 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")