Change gemma repo to lightricks one that is not gated

This commit is contained in:
Jaret Burkett 2026-03-23 11:00:32 -06:00
parent 253cb31362
commit e91827f9be
1 changed files with 1 additions and 1 deletions

View File

@ -77,7 +77,7 @@ dit_prefix = "model.diffusion_model."
vae_prefix = "vae."
audio_vae_prefix = "audio_vae."
vocoder_prefix = "vocoder."
base_te_path = "google/gemma-3-12b-it-qat-q4_0-unquantized"
base_te_path = "Lightricks/gemma-3-12b-it-qat-q4_0-unquantized"
HF_TOKEN = os.getenv("HF_TOKEN", None)