Update README.md
Browse files
README.md
CHANGED
|
@@ -59,7 +59,6 @@ bnb_config = BitsAndBytesConfig(
|
|
| 59 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 60 |
model = AutoModelForCausalLM.from_pretrained(
|
| 61 |
model_name,
|
| 62 |
-
revision="main", #qlora-merged (qLoRA finetuned for 3 epochs) or main (full parameter finetune for 1 epoch)
|
| 63 |
torch_dtype=torch.bfloat16,
|
| 64 |
trust_remote_code=True,
|
| 65 |
quantization_config=bnb_config,
|
|
|
|
| 59 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 60 |
model = AutoModelForCausalLM.from_pretrained(
|
| 61 |
model_name,
|
|
|
|
| 62 |
torch_dtype=torch.bfloat16,
|
| 63 |
trust_remote_code=True,
|
| 64 |
quantization_config=bnb_config,
|