Spaces:
Running
on
T4
Running
on
T4
simonduerr
commited on
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"]
|