habdine
/

Esm2Text-Base-v1-1

Text Generation

feature-extraction

Causal Language Modeling

Model card Files Files and versions

habdine commited on Oct 23

Commit

97498de

·

verified ·

1 Parent(s): dc6bdca

Update utils.py

Files changed (1) hide show

utils.py +0 -1

utils.py CHANGED Viewed

@@ -3,7 +3,6 @@ from transformers.models.gpt2.modeling_gpt2 import GPT2Attention, GPT2MLP
 from typing import Optional, Tuple, Union, Any, Dict, List
 from transformers import GPT2LMHeadModel
 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 typing import Optional, Tuple, Union, Any, Dict, List
 from transformers import GPT2LMHeadModel
 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