add push_to_hub to pipeline (#29172)
* add `push_to_hub` to pipeline * fix docs * format with ruff * update save_pretrained * update save_pretrained * remove unnecessary comment * switch to push_to_hub method in DynamicPipelineTester * remove unused imports * update docs for add_new_pipeline * fix docs for add_new_pipeline * add comment * fix italien docs * changes to token retrieval for pipelines * Update src/transformers/pipelines/base.py Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com> --------- Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>
This commit is contained in:
@@ -22,7 +22,7 @@ from pathlib import Path
|
||||
|
||||
import datasets
|
||||
import numpy as np
|
||||
from huggingface_hub import HfFolder, Repository, create_repo, delete_repo
|
||||
from huggingface_hub import HfFolder, delete_repo
|
||||
from requests.exceptions import HTTPError
|
||||
|
||||
from transformers import (
|
||||
@@ -846,9 +846,6 @@ class DynamicPipelineTester(unittest.TestCase):
|
||||
model = BertForSequenceClassification(config).eval()
|
||||
|
||||
with tempfile.TemporaryDirectory() as tmp_dir:
|
||||
create_repo(f"{USER}/test-dynamic-pipeline", token=self._token)
|
||||
repo = Repository(tmp_dir, clone_from=f"{USER}/test-dynamic-pipeline", token=self._token)
|
||||
|
||||
vocab_file = os.path.join(tmp_dir, "vocab.txt")
|
||||
with open(vocab_file, "w", encoding="utf-8") as vocab_writer:
|
||||
vocab_writer.write("".join([x + "\n" for x in self.vocab_tokens]))
|
||||
@@ -860,7 +857,7 @@ class DynamicPipelineTester(unittest.TestCase):
|
||||
del PIPELINE_REGISTRY.supported_tasks["pair-classification"]
|
||||
|
||||
classifier.save_pretrained(tmp_dir)
|
||||
# checks
|
||||
# checks if the configuration has been added after calling the save_pretrained method
|
||||
self.assertDictEqual(
|
||||
classifier.model.config.custom_pipelines,
|
||||
{
|
||||
@@ -871,8 +868,8 @@ class DynamicPipelineTester(unittest.TestCase):
|
||||
}
|
||||
},
|
||||
)
|
||||
|
||||
repo.push_to_hub()
|
||||
# use push_to_hub method to push the pipeline
|
||||
classifier.push_to_hub(f"{USER}/test-dynamic-pipeline", token=self._token)
|
||||
|
||||
# Fails if the user forget to pass along `trust_remote_code=True`
|
||||
with self.assertRaises(ValueError):
|
||||
|
||||
Reference in New Issue
Block a user