Skip to content

Commit 51fb06b

Browse files
authored
gemma_300m_lora variant is missing in Variant (#555)
fixing this as it will throw an annoying log every time the model is used.
2 parents 5754148 + a4b2db0 commit 51fb06b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/openpi/models/gemma.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ class Config:
5252
lora_configs: dict[str, lora.LoRAConfig] = dataclasses.field(default_factory=dict)
5353

5454

55-
Variant = Literal["dummy", "gemma_300m", "gemma_2b", "gemma_2b_lora"]
55+
Variant = Literal["dummy", "gemma_300m", "gemma_300m_lora", "gemma_2b", "gemma_2b_lora"]
5656

5757

5858
def get_config(variant: Variant) -> Config:

0 commit comments

Comments
 (0)