Commit
·
af0db75
1
Parent(s):
df17723
add pretrained
Browse files- custom_model.py +5 -1
- model.safetensors +3 -0
custom_model.py
CHANGED
|
@@ -20,4 +20,8 @@ class CustomModel(PreTrainedModel):
|
|
| 20 |
return output
|
| 21 |
|
| 22 |
AutoConfig.register("custom-model", CustomModelConfig)
|
| 23 |
-
AutoModel.register(CustomModelConfig, CustomModel)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 20 |
return output
|
| 21 |
|
| 22 |
AutoConfig.register("custom-model", CustomModelConfig)
|
| 23 |
+
AutoModel.register(CustomModelConfig, CustomModel)
|
| 24 |
+
|
| 25 |
+
# config = CustomModelConfig(hidden_size=128)
|
| 26 |
+
# model = CustomModel(config)
|
| 27 |
+
# model.save_pretrained("./")
|
model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:00e446dcb7abadbce1547c6379d8378f6ad20844a188d5682eac8ab4a08d7bad
|
| 3 |
+
size 66232
|