Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Commit
·
4c85687
1
Parent(s):
cf2dead
osc to omc
Browse files- 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/
|
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:
|