habdine
/

Prot2Text-Base-v1-1

Text Generation

feature-extraction

Causal Language Modeling

Model card Files Files and versions

habdine commited on 13 days ago

Commit

1341ae9

·

verified ·

1 Parent(s): afd1c75

Update utils.py

Files changed (1) hide show

utils.py +0 -1

utils.py CHANGED Viewed

@@ -4,7 +4,6 @@ from typing import Optional, Tuple, Union, Any, Dict, List
 from transformers import Seq2SeqTrainer, GPT2LMHeadModel
 from torch.utils.data.distributed import DistributedSampler
 import torch
-from transformers.deepspeed import is_deepspeed_zero3_enabled
 from transformers.generation.logits_process import LogitsProcessorList
 from transformers.generation.stopping_criteria import StoppingCriteriaList
 from transformers.generation.utils import GreedySearchOutput, GreedySearchEncoderDecoderOutput, BeamSearchOutput, BeamSearchEncoderDecoderOutput

 from transformers import Seq2SeqTrainer, GPT2LMHeadModel
 from torch.utils.data.distributed import DistributedSampler
 import torch
 from transformers.generation.logits_process import LogitsProcessorList
 from transformers.generation.stopping_criteria import StoppingCriteriaList
 from transformers.generation.utils import GreedySearchOutput, GreedySearchEncoderDecoderOutput, BeamSearchOutput, BeamSearchEncoderDecoderOutput