fix import for fresh transformers
This commit is contained in:
@@ -38,6 +38,8 @@ from transformers.modeling_outputs import (
|
||||
)
|
||||
from transformers.modeling_utils import (
|
||||
PreTrainedModel,
|
||||
)
|
||||
from transformers.pytorch_utils import (
|
||||
apply_chunking_to_forward,
|
||||
find_pruneable_heads_and_indices,
|
||||
prune_linear_layer,
|
||||
|
||||
+1
-1
@@ -1,3 +1,3 @@
|
||||
huggingface_hub
|
||||
pillow
|
||||
transformers
|
||||
transformers>=4.57.0
|
||||
Reference in New Issue
Block a user