Skip to content

Commit 5875c00

Browse files
gemma_300m_lora variant is missing in Variant
1 parent 072217e commit 5875c00

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)