V2 runpod - #2

Open

star2vec wants to merge 5 commits into

mainfrom

Commits on Apr 9, 2026

Fix unit tests and lock PyTorch to 2.4.1
Ecaterina
committed
a4141bd

Commits on Apr 10, 2026

Fix: move tensors to device and handle inference mode
Ecaterina
committed
827efba
Fix: set bnb_4bit_compute_dtype to float16 for massive speedup
Ecaterina
committed
42ab29d
Fix: cast training latents to float32 to match adapter
Ecaterina
committed
26d7fb7
Checkpoint: Fully functional Teacher-Student pipeline with fixed precision casting
star2vec
committed
6cbf729