SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Revert "Fix gradient checkpointing + fp16 autocast for most models" (#24420)

Browse Source 
Revert "Fix gradient checkpointing + fp16 autocast for most models (#24247)"

This reverts commit 285a48011d.
This commit is contained in:
Younes Belkada
2023-06-22 16:11:27 +02:00
committed by GitHub
parent ebb62e8880
commit 3ce3385c47
179 changed files with 271 additions and 836 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/transformers/models/sam/modeling_sam.py
View File

@@ -28,7 +28,6 @@ from torch import Tensor, nn
from ...activations import ACT2FN
from ...modeling_outputs import BaseModelOutput
from ...modeling_utils import PreTrainedModel
from ...pytorch_utils import torch_custom_checkpointing
from ...utils import ModelOutput, add_start_docstrings, add_start_docstrings_to_model_forward, logging
from .configuration_sam import SamConfig, SamMaskDecoderConfig, SamPromptEncoderConfig, SamVisionConfig
@@ -1050,7 +1049,7 @@ class SamVisionEncoder(nn.Module):
return custom_forward
layer_outputs = torch_custom_checkpointing(
layer_outputs = torch.utils.checkpoint.checkpoint(
create_custom_forward(layer_module),
hidden_states,
)