nit
This commit is contained in:
parent
e1b0262a9e
commit
d7355db6ae
|
@ -20,6 +20,7 @@ import torch
|
|||
import torch.utils.checkpoint
|
||||
from torch import nn
|
||||
from torch.nn import CrossEntropyLoss
|
||||
|
||||
from transformers import PretrainedConfig
|
||||
from transformers.models.llama.modeling_llama import (
|
||||
LlamaForCausalLM,
|
||||
|
|
Loading…
Reference in New Issue