MoraxCheng commited on
Commit
77cfeaa
·
1 Parent(s): ac2c54a

Update model selection for HF Spaces to use Large model and add local settings configuration

Browse files
Files changed (1) hide show
  1. app.py +1 -3
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
- # For HF Spaces, we recommend using Medium model due to memory constraints
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