We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 8fc84c6 commit d04013cCopy full SHA for d04013c
src/optimum/nvidia/models/auto.py
@@ -38,7 +38,7 @@ class AutoModelForCausalLM(ModelHubMixin):
38
"mixtral": LlamaForCausalLM,
39
"gemma": GemmaForCausalLM,
40
"qwen": QwenForCausalLM,
41
- "qwen2": QwenForCausalLM
+ "qwen2": QwenForCausalLM,
42
}
43
44
def __init__(self):
0 commit comments