saymrwulf/transformers
1
0
Fork 0
mirror of https://github.com/saymrwulf/transformers.git synced 2026-05-14 20:58:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[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"]