Merge pull request #57 from arjunguha/main

Removed unused line

Author: ArmelRandy

finetune/merge_peft_adapters.py

@@ -25,7 +25,6 @@ def main():
     model = PeftModel.from_pretrained(base_model, args.peft_model_path)
     model = model.merge_and_unload()
 
-    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     tokenizer = AutoTokenizer.from_pretrained(args.base_model_name_or_path)
 
     if args.push_to_hub:
@@ -38,4 +37,4 @@ def main():
         print(f"Model saved to {args.base_model_name_or_path}-merged")
 
 if __name__ == "__main__" :
-    main()
+    main()