kingJulio commited on
Commit
e0e030b
·
verified ·
1 Parent(s): e74eabb

Upload folder using huggingface_hub

Browse files
Files changed (3) hide show
  1. README.md +52 -0
  2. adapter_config.json +50 -0
  3. adapter_model.safetensors +3 -0
README.md ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Coda-Robotics/OpenVLA-ER-Select-Book-LoRA
2
+
3
+ ## Model Description
4
+
5
+ This is a LoRA adapter weights only (requires base OpenVLA model) of OpenVLA, fine-tuned on the select_book dataset.
6
+
7
+ ## Training Details
8
+
9
+ - **Dataset:** select_book
10
+ - **Number of Episodes:** 479
11
+ - **Batch Size:** 8
12
+ - **Training Steps:** 20000
13
+ - **Learning Rate:** 2e-5
14
+ - **LoRA Configuration:**
15
+ - Rank: 32
16
+ - Dropout: 0.0
17
+ - Target Modules: all-linear
18
+
19
+ ## Usage
20
+
21
+ ```python
22
+ from transformers import AutoProcessor, AutoModelForVision2Seq
23
+
24
+ # Load the model and processor
25
+ processor = AutoProcessor.from_pretrained("Coda-Robotics/OpenVLA-ER-Select-Book-LoRA")
26
+ model = AutoModelForVision2Seq.from_pretrained("Coda-Robotics/OpenVLA-ER-Select-Book-LoRA")
27
+
28
+ # Process an image
29
+ image = ... # Load your image
30
+ inputs = processor(images=image, return_tensors="pt")
31
+ outputs = model.generate(**inputs)
32
+ text = processor.decode(outputs[0], skip_special_tokens=True)
33
+ ```
34
+
35
+
36
+ ## Using with PEFT
37
+
38
+ To use this adapter with the base OpenVLA model:
39
+
40
+ ```python
41
+ from transformers import AutoProcessor, AutoModelForVision2Seq
42
+ from peft import PeftModel, PeftConfig
43
+
44
+ # Load the base model
45
+ base_model = AutoModelForVision2Seq.from_pretrained("openvla/openvla-7b")
46
+
47
+ # Load the LoRA adapter
48
+ adapter_model = PeftModel.from_pretrained(base_model, "{model_name}")
49
+
50
+ # Merge weights for faster inference (optional)
51
+ merged_model = adapter_model.merge_and_unload()
52
+ ```
adapter_config.json ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alpha_pattern": {},
3
+ "auto_mapping": {
4
+ "base_model_class": "OpenVLAForActionPrediction",
5
+ "parent_library": "transformers_modules.openvla.openvla-7b.31f090d05236101ebfc381b61c674dd4746d4ce0.modeling_prismatic"
6
+ },
7
+ "base_model_name_or_path": "openvla/openvla-7b",
8
+ "bias": "none",
9
+ "corda_config": null,
10
+ "eva_config": null,
11
+ "exclude_modules": null,
12
+ "fan_in_fan_out": false,
13
+ "inference_mode": true,
14
+ "init_lora_weights": "gaussian",
15
+ "layer_replication": null,
16
+ "layers_pattern": null,
17
+ "layers_to_transform": null,
18
+ "loftq_config": {},
19
+ "lora_alpha": 16,
20
+ "lora_bias": false,
21
+ "lora_dropout": 0.0,
22
+ "megatron_config": null,
23
+ "megatron_core": "megatron.core",
24
+ "modules_to_save": null,
25
+ "peft_type": "LORA",
26
+ "r": 32,
27
+ "rank_pattern": {},
28
+ "revision": null,
29
+ "target_modules": [
30
+ "gate_proj",
31
+ "v_proj",
32
+ "qkv",
33
+ "attn.proj",
34
+ "o_proj",
35
+ "q_proj",
36
+ "fc1",
37
+ "k_proj",
38
+ "attn_pool.proj",
39
+ "fc2",
40
+ "down_proj",
41
+ "fc3",
42
+ "up_proj",
43
+ "q",
44
+ "kv"
45
+ ],
46
+ "task_type": null,
47
+ "trainable_token_indices": null,
48
+ "use_dora": false,
49
+ "use_rslora": false
50
+ }
adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bbafa2881d7852f9ae8e5ef39d49f394fb7d08be1b211f9cfbcdadf8cae87083
3
+ size 438387152