paligemma2-docci / src /config.py
sitammeur's picture
Update src/config.py
15acea6 verified
raw
history blame contribute delete
82 Bytes
# Model settings
device = "cuda"
model_name = "google/paligemma2-3b-ft-docci-448"