wassname/peft
1
0
Fork 0
mirror of https://github.com/wassname/peft.git synced 2026-09-10 12:20:21 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix CI

Browse Source
This commit is contained in:
younesbelkada
2023-04-03 16:47:09 +00:00
parent c2e9a6681a
commit 2fe22da3a2
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_common_gpu.py
+1 -1
View File
@@ -20,7 +20,7 @@ import torch
from transformers import AutoModelForCausalLM, AutoModelForSeq2SeqLM, AutoTokenizer, WhisperForConditionalGeneration
from peft import LoraConfig, PeftModel, get_peft_model
from peft.utils import is_bnb_available
from peft.import_utils import is_bnb_available
from .testing_utils import require_bitsandbytes, require_torch_gpu, require_torch_multi_gpu