From 8c83386ef413174bb2ee7167d1d13fa0fe00516c Mon Sep 17 00:00:00 2001
From: younesbelkada <younesbelkada@gmail.com>
Date: Tue, 4 Apr 2023 08:37:32 +0000
Subject: [PATCH] few fixes

---
 .../int8_training/fine_tune_blip2_int8.py     | 21 +++++++++++++++++--
 src/peft/tuners/lora.py                       |  2 --
 2 files changed, 19 insertions(+), 4 deletions(-)

diff --git a/examples/int8_training/fine_tune_blip2_int8.py b/examples/int8_training/fine_tune_blip2_int8.py
index 1e73ffb..25121f6 100644
--- a/examples/int8_training/fine_tune_blip2_int8.py
+++ b/examples/int8_training/fine_tune_blip2_int8.py
@@ -1,3 +1,17 @@
+# coding=utf-8
+# Copyright 2023-present the HuggingFace Inc. team.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
 import torch
 from datasets import load_dataset
 from torch.utils.data import DataLoader, Dataset
@@ -6,6 +20,7 @@ from transformers import AutoModelForVision2Seq, AutoProcessor
 from peft import LoraConfig, get_peft_model
 
 
+# Let's define the LoraConfig
 config = LoraConfig(
     r=16,
     lora_alpha=32,
@@ -14,12 +29,15 @@ config = LoraConfig(
     task_type="VISION_2_SEQ",
 )
 
+# We load our model and processor using `transformers`
 model = AutoModelForVision2Seq.from_pretrained("Salesforce/blip2-opt-2.7b", load_in_8bit=True, device_map={"": 0})
 processor = AutoProcessor.from_pretrained("Salesforce/blip2-opt-2.7b")
-model = get_peft_model(model, config)
 
+# Get our peft model and print the number of trainable parameters
+model = get_peft_model(model, config)
 model.print_trainable_parameters()
 
+# Let's load the dataset here!
 dataset = load_dataset("ybelkada/football-dataset", split="train")
 
 
@@ -61,7 +79,6 @@ train_dataloader = DataLoader(train_dataset, shuffle=True, batch_size=2, collate
 optimizer = torch.optim.AdamW(model.parameters(), lr=5e-5)
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-model.to(device)
 
 model.train()
 
diff --git a/src/peft/tuners/lora.py b/src/peft/tuners/lora.py
index 9745467..51cd56f 100644
--- a/src/peft/tuners/lora.py
+++ b/src/peft/tuners/lora.py
@@ -405,8 +405,6 @@ class Linear(nn.Linear, LoraLayer):
         elif self.r > 0 and not self.merged:
             result = F.linear(x, transpose(self.weight, self.fan_in_fan_out), bias=self.bias)
             if self.r > 0:
-                x = x.to(self.lora_A.weight.dtype)
-
                 result += self.lora_B(self.lora_A(self.lora_dropout(x))) * self.scaling
             return result
         else: