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
|