Skip to content
Snippets Groups Projects
Commit f058ff6c authored by Hamid Shojanazeri's avatar Hamid Shojanazeri
Browse files

update due to peft new release

parent 6a7478a6
No related branches found
No related tags found
No related merge requests found
......@@ -9,7 +9,7 @@ import fire
import random
import torch
import torch.optim as optim
from peft import get_peft_model, prepare_model_for_int8_training
from peft import get_peft_model, prepare_model_for_kbit_training
from torch.distributed.fsdp import (
FullyShardedDataParallel as FSDP,
ShardingStrategy
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment