paligemma-docci / src /config.py
sitammeur's picture
Update src/config.py
6df4743 verified
raw
history blame contribute delete
78 Bytes
# Model settings
device = "cuda"
model_name = "google/paligemma2-10b-mix-448"