Wonderplex commited on
Commit
7438440
1 Parent(s): 757f773

Bug/hfspace (#38)

Browse files

* added format promp and changed sotopia information accordion

* added attributes for lfs

* removed local adpaters

Files changed (2) hide show
  1. app.py +1 -1
  2. sotopia_pi_adapter/adapter_config.json +0 -21
app.py CHANGED
@@ -58,7 +58,7 @@ def prepare(model_name):
58
  )
59
  )
60
  tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.1")
61
- model = PeftModel.from_pretrained(model, "./sotopia_pi_adapter").to("cuda")
62
  else:
63
  raise RuntimeError(f"Model {model_name} not supported")
64
  return model, tokenizer
 
58
  )
59
  )
60
  tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.1")
61
+ model = PeftModel.from_pretrained(model, model_name).to("cuda")
62
  else:
63
  raise RuntimeError(f"Model {model_name} not supported")
64
  return model, tokenizer
sotopia_pi_adapter/adapter_config.json DELETED
@@ -1,21 +0,0 @@
1
- {
2
- "auto_mapping": null,
3
- "base_model_name_or_path": "mistralai/Mistral-7B-Instruct-v0.1",
4
- "bias": "none",
5
- "fan_in_fan_out": false,
6
- "inference_mode": true,
7
- "init_lora_weights": true,
8
- "layers_pattern": null,
9
- "layers_to_transform": null,
10
- "lora_alpha": 32.0,
11
- "lora_dropout": 0.1,
12
- "modules_to_save": null,
13
- "peft_type": "LORA",
14
- "r": 8,
15
- "revision": null,
16
- "target_modules": [
17
- "q_proj",
18
- "v_proj"
19
- ],
20
- "task_type": "CAUSAL_LM"
21
- }