Spaces:
Running
on
T4
Running
on
T4
Commit
·
7b9daf5
1
Parent(s):
fde9ec3
Update app.py
Browse files
app.py
CHANGED
|
@@ -276,7 +276,7 @@ def setup_proteinmpnn(model_name="vanilla—v_48_020", backbone_noise=0.00):
|
|
| 276 |
if model_folder_path[-1] != "/":
|
| 277 |
model_folder_path = model_folder_path + "/"
|
| 278 |
checkpoint_path = model_folder_path + f"{model_name}.pt"
|
| 279 |
-
|
| 280 |
checkpoint = torch.load(checkpoint_path, map_location=device)
|
| 281 |
|
| 282 |
noise_level_print = checkpoint["noise_level"]
|
|
|
|
| 276 |
if model_folder_path[-1] != "/":
|
| 277 |
model_folder_path = model_folder_path + "/"
|
| 278 |
checkpoint_path = model_folder_path + f"{model_name}.pt"
|
| 279 |
+
print("using ProteinMPNN weights from: ", checkpoint_path)
|
| 280 |
checkpoint = torch.load(checkpoint_path, map_location=device)
|
| 281 |
|
| 282 |
noise_level_print = checkpoint["noise_level"]
|