Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
e85bdfa
1
Parent(s):
f4cb90c
try to fix bug
Browse files
utils.py
CHANGED
@@ -205,6 +205,7 @@ class MolecularPropertyPredictionModel():
|
|
205 |
try:
|
206 |
self.adapter_name = adapter_name
|
207 |
self.lora_model = PeftModel.from_pretrained(self.base_model, adapter_id, token = os.environ.get("TOKEN"))
|
|
|
208 |
if adapter_name not in self.apapter_scaler_path:
|
209 |
self.apapter_scaler_path[adapter_name] = hf_hub_download(adapter_id, filename="scaler.pkl", token = os.environ.get("TOKEN"))
|
210 |
if os.path.exists(self.apapter_scaler_path[adapter_name]):
|
|
|
205 |
try:
|
206 |
self.adapter_name = adapter_name
|
207 |
self.lora_model = PeftModel.from_pretrained(self.base_model, adapter_id, token = os.environ.get("TOKEN"))
|
208 |
+
self.lora_model.to("cuda")
|
209 |
if adapter_name not in self.apapter_scaler_path:
|
210 |
self.apapter_scaler_path[adapter_name] = hf_hub_download(adapter_id, filename="scaler.pkl", token = os.environ.get("TOKEN"))
|
211 |
if os.path.exists(self.apapter_scaler_path[adapter_name]):
|