zulissimeta commited on
Commit
4c85687
·
1 Parent(s): cf2dead

osc to omc

Browse files
Files changed (1) hide show
  1. hf_calculator.py +1 -3
hf_calculator.py CHANGED
@@ -49,7 +49,7 @@ class HFEndpointCalculator(Calculator):
49
  if not example:
50
  try:
51
  hf_hub.HfApi().auth_check(
52
- repo_id="facebook/OMAT24", token=oauth_token.token
53
  )
54
  hash_save_file(atoms, task_name, "/data/custom_inputs/")
55
  except (hf_hub.errors.HfHubHTTPError, AttributeError):
@@ -69,8 +69,6 @@ class HFEndpointCalculator(Calculator):
69
  Calculator.calculate(self, atoms, properties, system_changes)
70
 
71
  task_name = self.task_name.lower()
72
- if task_name == "omc":
73
- task_name = "osc"
74
 
75
  # Run inference via a post request to the endpoint
76
  try:
 
49
  if not example:
50
  try:
51
  hf_hub.HfApi().auth_check(
52
+ repo_id="facebook/UMA", token=oauth_token.token
53
  )
54
  hash_save_file(atoms, task_name, "/data/custom_inputs/")
55
  except (hf_hub.errors.HfHubHTTPError, AttributeError):
 
69
  Calculator.calculate(self, atoms, properties, system_changes)
70
 
71
  task_name = self.task_name.lower()
 
 
72
 
73
  # Run inference via a post request to the endpoint
74
  try: