Views
No views yet
| Metric | Value |
|---|---|
| Eval Loss | 2.2045 |
| Eval Model Preparation Time | 0.0197 |
| Eval Runtime | 94.6703 |
| Eval Samples Per Second | 8.8730 |
| Eval Steps Per Second | 1.1090 |
1import torch
2from transformers import AutoProcessor, AutoModelForImageTextToText
3from peft import PeftModel
4
5base_model_id = "google/medgemma-4b-it"
6adapter_id = "mnbvcxzz4869/medgemma-tb-it-lora-tb"
7
8processor = AutoProcessor.from_pretrained(base_model_id)
9
10model = AutoModelForImageTextToText.from_pretrained(
11 base_model_id,
12 torch_dtype=torch.bfloat16,
13 device_map="auto"
14)
15
16model = PeftModel.from_pretrained(model, adapter_id)