Manhph2211/PulseLM

PulseLM: A Foundation Dataset and Benchmark for PPG-Text Learning

Hung Manh Pham*   Jinyang Wu*   Xiao Ma   Yiming Zhang   Yixin Xu   Aaqib Saeed   Bin Zhu†   Zhou Pan†   Dong Ma†

* Equal contribution    † Corresponding authors

Quick Start

from transformers import AutoModelForCausalLM, AutoTokenizer
import torch
import neurokit2 as nk

model = AutoModelForCausalLM.from_pretrained(
    "Manhph2211/PulseLM",
    trust_remote_code=True,
    dtype=torch.bfloat16,
    device_map="auto",
)
tokenizer = AutoTokenizer.from_pretrained("Manhph2211/PulseLM", trust_remote_code=True)
model.eval()