LesterCerioli commited on
Commit
2f6381c
·
1 Parent(s): b08433f

Building first version from Golang LLM

Browse files
src/llm_go/deployment/hf_uploader.py CHANGED
@@ -98,7 +98,7 @@ class HuggingFaceUploader:
98
  mc = ModelConfig.load(ckpt / "config.json")
99
 
100
  hf_cfg = {
101
- "architectures": ["GoLLMForCausalLM"],
102
  "model_type": "gpt2",
103
  "vocab_size": mc.vocab_size,
104
  "n_positions": mc.max_seq_len,
 
98
  mc = ModelConfig.load(ckpt / "config.json")
99
 
100
  hf_cfg = {
101
+ "architectures": ["GPT2LMHeadModel"],
102
  "model_type": "gpt2",
103
  "vocab_size": mc.vocab_size,
104
  "n_positions": mc.max_seq_len,