Skip to content

Commit ed63133

Browse files
committed
🐛 Fix converting of peft model to type errorneously
Signed-off-by: gkumbhat <kumbhat.gaurav@gmail.com>
1 parent b4133d1 commit ed63133

1 file changed

Lines changed: 0 additions & 1 deletion

File tree

caikit_nlp/modules/text_generation/peft_prompt_tuning.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -414,7 +414,6 @@ def train(
414414
# Convert our Peft model (not just the underlying
415415
# transformers model) to the right underlying type.
416416
device = cls._get_device(device)
417-
cls.convert_peft_model_to_type(device, peft_model, torch_dtype)
418417

419418
## Generate data loader from stream
420419
training_dataset: Union[

0 commit comments

Comments
 (0)