Upload 4 files

README.md

@@ -1,3 +1,39 @@
----
-license: apache-2.0
----
+
+---
+
+base_model: ProTrekHub/Protein_Encoder_35M 
+
+library_name: peft
+
+---
+
+
+
+# Model Card for Model-Demo-ProTrek-35M
+<slot name='description'>
+
+## Task type
+Protein-level Classification
+
+## Model input type
+AA Sequence
+
+## LoRA config
+
+- **r:** 8
+- **lora_dropout:** 0.0
+- **lora_alpha:** 16
+- **target_modules:** ['key', 'value', 'output.dense', 'query', 'intermediate.dense']
+- **modules_to_save:** ['classifier']
+
+## Training config
+
+- **optimizer:**
+  - **class:** AdamW
+  - **betas:** (0.9, 0.98)
+  - **weight_decay:** 0.01
+- **learning rate:** 0.0005
+- **epoch:** 5
+- **batch size:** 1
+- **precision:** 16-mixed 
+

adapter_config.json

@@ -0,0 +1,34 @@
+{
+  "alpha_pattern": {},
+  "auto_mapping": null,
+  "base_model_name_or_path": "ProTrekHub/Protein_Encoder_35M",
+  "bias": "none",
+  "fan_in_fan_out": false,
+  "inference_mode": true,
+  "init_lora_weights": true,
+  "layer_replication": null,
+  "layers_pattern": null,
+  "layers_to_transform": null,
+  "loftq_config": {},
+  "lora_alpha": 16,
+  "lora_dropout": 0.0,
+  "megatron_config": null,
+  "megatron_core": "megatron.core",
+  "modules_to_save": [
+    "classifier"
+  ],
+  "peft_type": "LORA",
+  "r": 8,
+  "rank_pattern": {},
+  "revision": null,
+  "target_modules": [
+    "key",
+    "value",
+    "output.dense",
+    "query",
+    "intermediate.dense"
+  ],
+  "task_type": "SEQ_CLS",
+  "use_dora": false,
+  "use_rslora": false
+}

adapter_model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6fe056a85c468d0eef54a06474f7b8661befc4ec2d082288be5f69f200c2f136
+size 4265400

metadata.json

@@ -0,0 +1,4 @@
+{
+    "training_task_type": "Protein-level Classification",
+    "training_data_type": "AA"
+}