Spaces:
Paused
Paused
Tao Wu
commited on
Commit
·
871dabe
1
Parent(s):
94129fe
8bit
Browse files- app/embedding_setup.py +2 -6
app/embedding_setup.py
CHANGED
|
@@ -53,9 +53,7 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
| 53 |
|
| 54 |
rec_adapter = PeftModel.from_pretrained(
|
| 55 |
model,
|
| 56 |
-
lora_weights_rec
|
| 57 |
-
torch_dtype=torch.float16,
|
| 58 |
-
device_map={'': 0}
|
| 59 |
)
|
| 60 |
|
| 61 |
|
|
@@ -148,9 +146,7 @@ def compare_docs_with_context(doc_a, doc_b, target_occupation_name, target_occup
|
|
| 148 |
#-----------------------------------------explanation-------------------------------------
|
| 149 |
exp_adapter = PeftModel.from_pretrained(
|
| 150 |
model,
|
| 151 |
-
lora_weights_exp
|
| 152 |
-
torch_dtype=torch.float16,
|
| 153 |
-
device_map={'': 0}
|
| 154 |
)
|
| 155 |
exp_adapter.eval()
|
| 156 |
|
|
|
|
| 53 |
|
| 54 |
rec_adapter = PeftModel.from_pretrained(
|
| 55 |
model,
|
| 56 |
+
lora_weights_rec
|
|
|
|
|
|
|
| 57 |
)
|
| 58 |
|
| 59 |
|
|
|
|
| 146 |
#-----------------------------------------explanation-------------------------------------
|
| 147 |
exp_adapter = PeftModel.from_pretrained(
|
| 148 |
model,
|
| 149 |
+
lora_weights_exp
|
|
|
|
|
|
|
| 150 |
)
|
| 151 |
exp_adapter.eval()
|
| 152 |
|