mirror of
https://github.com/wassname/peft.git
synced 2026-10-29 13:11:48 +08:00
quality
This commit is contained in:
1 parent
65112b75bb
commit
73cd16b7b5
1 file changed
+1
-1
@@ -19,8 +19,8 @@ import warnings
|
||||
|
||||
import torch
|
||||
from accelerate import dispatch_model, infer_auto_device_map
|
||||
from accelerate.utils import get_balanced_memory
|
||||
from accelerate.hooks import AlignDevicesHook, add_hook_to_module, remove_hook_from_submodules
|
||||
from accelerate.utils import get_balanced_memory
|
||||
from torch.nn import BCEWithLogitsLoss, CrossEntropyLoss, MSELoss
|
||||
from transformers import PreTrainedModel
|
||||
from transformers.modeling_outputs import SequenceClassifierOutput, TokenClassifierOutput
|
||||
|
||||
Reference in new issue
Block a user