czczup commited on
Commit
0fbe95a
1 Parent(s): 4076c24

Update modeling_internlm2.py

Browse files
Files changed (1) hide show
  1. modeling_internlm2.py +0 -2
modeling_internlm2.py CHANGED
@@ -42,10 +42,8 @@ except: # noqa # pylint: disable=bare-except
42
  from typing import Any, List, Optional, Tuple, Union
43
  import torch.distributed as dist
44
  import torch.utils.checkpoint
45
- from peft import LoraConfig, get_peft_model
46
  from torch import nn
47
  from torch.nn import CrossEntropyLoss
48
- from transformers import GenerationConfig, LlamaForCausalLM, LlamaTokenizer
49
  from transformers.generation.logits_process import LogitsProcessorList
50
  from transformers.generation.stopping_criteria import StoppingCriteriaList
51
  from transformers.generation.streamers import BaseStreamer
 
42
  from typing import Any, List, Optional, Tuple, Union
43
  import torch.distributed as dist
44
  import torch.utils.checkpoint
 
45
  from torch import nn
46
  from torch.nn import CrossEntropyLoss
 
47
  from transformers.generation.logits_process import LogitsProcessorList
48
  from transformers.generation.stopping_criteria import StoppingCriteriaList
49
  from transformers.generation.streamers import BaseStreamer