Views
No views yet
1from transformers import AutoModelForSequenceClassification, AutoTokenizer
2
3# Load the tokenizer, adjust configuration if needed
4tokenizer = AutoTokenizer.from_pretrained(model_name)
5model = AutoModelForCausalLM.from_pretrained(model_name)
6
7# Load the fine-tuned model with its trained weights
8fine_tuned_model = AutoModelForSequenceClassification.from_pretrained(
9 'fionazhang/mistral_7b_environment',
10)
11
12# Now you can use `fine_tuned_model` for inference or further training
13input_text = "The impact of climate change on"
14output_text = fine_tuned_model.generate(tokenizer.encode(input_text, return_tensors="pt"))
15
16print(tokenizer.decode(output_text[0], skip_special_tokens=True))
171training_arguments = TrainingArguments(
2 output_dir="",
3 num_train_epochs=1,
4 per_device_train_batch_size=4,
5 gradient_accumulation_steps=1,
6 optim="paged_adamw_32bit",
7 save_steps=25,
8 logging_steps=25,
9 learning_rate=2e-4,
10 weight_decay=0.001,
11 fp16=False,
12 bf16=False,
13 max_grad_norm=0.3,
14 max_steps=-1,
15 warmup_ratio=0.03,
16 group_by_length=True,
17 lr_scheduler_type="constant",
18 report_to="wandb"
19)