p1atdev/MangaLineExtraction-hf

Apr 7

this pr will add pipeline to your repo to test it out before merging you can try the following code :

pip install loadimg

# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("image-to-image", model="p1atdev/MangaLineExtraction-hf", 
revision="refs/pr/1",trust_remote_code=True)

add pipeline6afbcc3d

not-lain

Apr 7

btw can you tag me in more AI models that have custom architectures too ? (yours or any other person)
i would also like to share with you something that the team in huggingface have been working on which is the PyTorchModelHubMixin which will allow you to add push_to_hub from_pretrained save_pretrained to any custom model

docs : https://huggingface.co/docs/huggingface_hub/main/en/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin
mini example : https://github.com/not-lain/PyTorchModelHubMixin-template

p1atdev

Owner Apr 10

I see. I'll mention you if I see any custom code.
Nice work👍

p1atdev changed pull request status to merged Apr 10

not-lain

Apr 10

@p1atdev thank youuu 🤗