diff --git a/merge.py b/merge.py index 56418e65a..f881802a4 100644 --- a/merge.py +++ b/merge.py @@ -31,7 +31,6 @@ model = LlamaForCausalLM.from_pretrained( load_in_8bit=False, torch_dtype=torch.float16, device_map={"": "cpu"}, - )