How to use from the
Use from the
PEFT library
from peft import PeftModel
from transformers import AutoModelForCausalLM

base_model = AutoModelForCausalLM.from_pretrained("TinyLlama/TinyLlama-1.1B-Chat-v1.0")
model = PeftModel.from_pretrained(base_model, "DIA-MVP/tinyllama-lora-a100")

TinyLlama 1.1B Chat โ€” LoRA (NVIDIA A100)

A demo model from the Data & Impact Accounting (DIA) lab. It performs instruction-tuning (LoRA adapter) via LoRA (PEFT), with the base model TinyLlama/TinyLlama-1.1B-Chat-v1.0, trained on NVIDIA A100.

The point of this repo is not the model itself but its dia_report โ€” a standardized record of the energy, carbon, and water used to train it, embedded in this card's metadata.

This footprint feeds the DIA dashboard, which rolls up a base model and all its derivatives to show the cumulative carbon, water, and energy cost of a model family.

Training footprint

Metric Value
Hardware 1ร— NVIDIA A100-SXM4-80GB
Compute 0.035 GPU-hours
Energy 0.0124 (measured) kWh
Carbon 0.0008 (measured) kgCOโ‚‚eq
Water 0.022โ€“0.05 (estimated-from-default-wue) L
Grid region ca-on

Energy and carbon are measured with CodeCarbon; water is estimated from a default water-usage-effectiveness range. Carbon uses the local grid's intensity (Ontario, ~0.03 kgCOโ‚‚eq/kWh).

Reproduce

REPO=DIA-MVP/tinyllama-lora-a100 python scripts/train_llama_lora.py

Links

Downloads last month
6
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support

Model tree for DIA-MVP/tinyllama-lora-a100

Adapter
(1,614)
this model