Spaces:
Running
Running
Update src/streamlit_app.py
Browse files- src/streamlit_app.py +1 -1
src/streamlit_app.py
CHANGED
|
@@ -941,7 +941,7 @@ SEVEN_NET_MODELS = {
|
|
| 941 |
}
|
| 942 |
@st.cache_resource
|
| 943 |
def get_mace_model(model_path, dispersion, device, selected_default_dtype):
|
| 944 |
-
return mace_mp(model=model_path, dispersion=
|
| 945 |
|
| 946 |
@st.cache_resource
|
| 947 |
def get_fairchem_model(selected_model_name, model_path_or_name, device, selected_task_type_fc): # Renamed args to avoid conflict
|
|
|
|
| 941 |
}
|
| 942 |
@st.cache_resource
|
| 943 |
def get_mace_model(model_path, dispersion, device, selected_default_dtype):
|
| 944 |
+
return mace_mp(model=model_path, dispersion=dispersion, device=device, default_dtype=selected_default_dtype)
|
| 945 |
|
| 946 |
@st.cache_resource
|
| 947 |
def get_fairchem_model(selected_model_name, model_path_or_name, device, selected_task_type_fc): # Renamed args to avoid conflict
|