Add support for ai-forever/mGPT model (#606)

SeuperHakkerJa · web-flow · commit 570a911d8425 · 2024-05-28T01:08:50.000+02:00
* add mGPT model path and alias

* reformat
diff --git a/transformer_lens/loading_from_pretrained.py b/transformer_lens/loading_from_pretrained.py
@@ -182,6 +182,7 @@
     "01-ai/Yi-34B",
     "01-ai/Yi-6B-Chat",
     "01-ai/Yi-34B-Chat",
+    "ai-forever/mGPT",
 ]
 """Official model names for models on HuggingFace."""
 
@@ -595,6 +596,7 @@
     "01-ai/Yi-34B": ["yi-34b", "Yi-34B"],
     "01-ai/Yi-6B-Chat": ["yi-6b-chat", "Yi-6B-Chat"],
     "01-ai/Yi-34B-Chat": ["yi-34b-chat", "Yi-34B-Chat"],
+    "ai-forever/mGPT": ["mGPT"],
 }
 """Model aliases for models on HuggingFace."""
 

Original file line number	Diff line number	Diff line change
`@@ -182,6 +182,7 @@`
`182`	`182`	`"01-ai/Yi-34B",`
`183`	`183`	`"01-ai/Yi-6B-Chat",`
`184`	`184`	`"01-ai/Yi-34B-Chat",`
	`185`	`+ "ai-forever/mGPT",`
`185`	`186`	`]`
`186`	`187`	`"""Official model names for models on HuggingFace."""`
`187`	`188`
`@@ -595,6 +596,7 @@`
`595`	`596`	`"01-ai/Yi-34B": ["yi-34b", "Yi-34B"],`
`596`	`597`	`"01-ai/Yi-6B-Chat": ["yi-6b-chat", "Yi-6B-Chat"],`
`597`	`598`	`"01-ai/Yi-34B-Chat": ["yi-34b-chat", "Yi-34B-Chat"],`
	`599`	`+ "ai-forever/mGPT": ["mGPT"],`
`598`	`600`	`}`
`599`	`601`	`"""Model aliases for models on HuggingFace."""`
`600`	`602`