Update modeling_internlm2.py
Browse files- 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
|