Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
77cfeaa
1
Parent(s):
ac2c54a
Update model selection for HF Spaces to use Large model and add local settings configuration
Browse files
app.py
CHANGED
@@ -246,9 +246,7 @@ def score_and_create_matrix_all_singles(sequence,mutation_range_start=None,mutat
|
|
246 |
model_path = download_model_from_hf("Tranception_Medium")
|
247 |
model = tranception.model_pytorch.TranceptionLMHeadModel.from_pretrained(pretrained_model_name_or_path=model_path)
|
248 |
elif model_type=="Large":
|
249 |
-
|
250 |
-
print("Note: Large model requires significant memory. Using Medium model for HF Spaces deployment.")
|
251 |
-
model_path = download_model_from_hf("Tranception_Medium")
|
252 |
model = tranception.model_pytorch.TranceptionLMHeadModel.from_pretrained(pretrained_model_name_or_path=model_path)
|
253 |
|
254 |
# Device selection - for HF Spaces, typically CPU
|
|
|
246 |
model_path = download_model_from_hf("Tranception_Medium")
|
247 |
model = tranception.model_pytorch.TranceptionLMHeadModel.from_pretrained(pretrained_model_name_or_path=model_path)
|
248 |
elif model_type=="Large":
|
249 |
+
model_path = download_model_from_hf("Tranception_Large")
|
|
|
|
|
250 |
model = tranception.model_pytorch.TranceptionLMHeadModel.from_pretrained(pretrained_model_name_or_path=model_path)
|
251 |
|
252 |
# Device selection - for HF Spaces, typically CPU
|