ereniko commited on
Commit
230fc1a
·
verified ·
1 Parent(s): 47bb023

Upload Ivme-Coder-v1 (Otter 1): safetensors + custom modeling code

Browse files
Files changed (3) hide show
  1. README.md +1 -1
  2. config.json +1 -1
  3. model.safetensors +2 -2
README.md CHANGED
@@ -64,7 +64,7 @@ import torch
64
  from transformers import AutoModelForCausalLM, AutoTokenizer
65
 
66
  model = AutoModelForCausalLM.from_pretrained(
67
- "IvmeLabs/Ivme-Coder-v1", trust_remote_code=True, dtype=torch.bfloat16,
68
  )
69
  tokenizer = AutoTokenizer.from_pretrained("IvmeLabs/Ivme-Coder-v1", trust_remote_code=True)
70
  model.eval()
 
64
  from transformers import AutoModelForCausalLM, AutoTokenizer
65
 
66
  model = AutoModelForCausalLM.from_pretrained(
67
+ "IvmeLabs/Ivme-Coder-v1", trust_remote_code=True, dtype=torch.float32,
68
  )
69
  tokenizer = AutoTokenizer.from_pretrained("IvmeLabs/Ivme-Coder-v1", trust_remote_code=True)
70
  model.eval()
config.json CHANGED
@@ -15,6 +15,6 @@
15
  "AutoConfig": "modeling_ivme_coder.IvmeCoderConfig",
16
  "AutoModelForCausalLM": "modeling_ivme_coder.IvmeCoderV1ForCausalLM"
17
  },
18
- "torch_dtype": "bfloat16",
19
  "transformers_version": "4.x"
20
  }
 
15
  "AutoConfig": "modeling_ivme_coder.IvmeCoderConfig",
16
  "AutoModelForCausalLM": "modeling_ivme_coder.IvmeCoderV1ForCausalLM"
17
  },
18
+ "torch_dtype": "float32",
19
  "transformers_version": "4.x"
20
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c4dd525d2983b01268b1eb7dfc2186386aae1ed83e9b4285106d8c5b159e81a3
3
- size 92038912
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cb47d80da9d26d9447718597e01c1619c35d850d1ad60b6d78d78e69db0e3b97
3
+ size 184018856