Spaces:
Running
on
Zero
Running
on
Zero
feiyang-cai
commited on
Update utils.py
Browse files
utils.py
CHANGED
@@ -204,8 +204,10 @@ class MolecularPropertyPredictionModel():
|
|
204 |
# switch adapter
|
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]):
|
|
|
204 |
# switch adapter
|
205 |
try:
|
206 |
self.adapter_name = adapter_name
|
207 |
+
print(self.adapter_name, adapter_id)
|
208 |
self.lora_model = PeftModel.from_pretrained(self.base_model, adapter_id, token = os.environ.get("TOKEN"))
|
209 |
self.lora_model.to("cuda")
|
210 |
+
print(self.lora_model)
|
211 |
if adapter_name not in self.apapter_scaler_path:
|
212 |
self.apapter_scaler_path[adapter_name] = hf_hub_download(adapter_id, filename="scaler.pkl", token = os.environ.get("TOKEN"))
|
213 |
if os.path.exists(self.apapter_scaler_path[adapter_name]):
|