riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-01 02:31:11 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

[fix] cannot import name 'Pop2PianoFeatureExtractor' from 'transformers' (#35604)

Browse Source 
* update pop2piano __init__

* add lib check

* update fix

* revert
This commit is contained in:
Fanli Lin 2025-01-20 22:21:45 +08:00 committed by GitHub
parent b80e334e71
commit 3998fa8aab
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/transformers/models/pop2piano/feature_extraction_pop2piano.py
View File

@ -448,3 +448,6 @@ class Pop2PianoFeatureExtractor(SequenceFeatureExtractor):
)
return output
__all__ = ["Pop2PianoFeatureExtractor"]

3
src/transformers/models/pop2piano/processing_pop2piano.py
View File

@ -137,3 +137,6 @@ class Pop2PianoProcessor(ProcessorMixin):
def from_pretrained(cls, pretrained_model_name_or_path, **kwargs):
args = cls._get_arguments_from_pretrained(pretrained_model_name_or_path, **kwargs)
return cls(*args)
__all__ = ["Pop2PianoProcessor"]

3
src/transformers/models/pop2piano/tokenization_pop2piano.py
View File

@ -714,3 +714,6 @@ class Pop2PianoTokenizer(PreTrainedTokenizer):
return BatchEncoding({"notes": notes_list, "pretty_midi_objects": pretty_midi_objects_list})
return BatchEncoding({"notes": notes_list})
__all__ = ["Pop2PianoTokenizer"]