We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e25ab3a commit 80a39c2Copy full SHA for 80a39c2
finetune/merge_peft_adapters.py
@@ -25,7 +25,6 @@ def main():
25
model = PeftModel.from_pretrained(base_model, args.peft_model_path)
26
model = model.merge_and_unload()
27
28
- device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
29
tokenizer = AutoTokenizer.from_pretrained(args.base_model_name_or_path)
30
31
if args.push_to_hub:
@@ -38,4 +37,4 @@ def main():
38
37
print(f"Model saved to {args.base_model_name_or_path}-merged")
39
40
if __name__ == "__main__" :
41
- main()
+ main()
0 commit comments