We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent b4133d1 commit ed63133Copy full SHA for ed63133
1 file changed
caikit_nlp/modules/text_generation/peft_prompt_tuning.py
@@ -414,7 +414,6 @@ def train(
414
# Convert our Peft model (not just the underlying
415
# transformers model) to the right underlying type.
416
device = cls._get_device(device)
417
- cls.convert_peft_model_to_type(device, peft_model, torch_dtype)
418
419
## Generate data loader from stream
420
training_dataset: Union[
0 commit comments