ManasSharma07 commited on
Commit
8b34e4c
·
verified ·
1 Parent(s): c703345

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +4 -2
src/streamlit_app.py CHANGED
@@ -1009,7 +1009,8 @@ if model_type == "MACE":
1009
  model_path = MACE_MODELS[selected_model]
1010
  if selected_model == "MACE OMAT Medium":
1011
  st.sidebar.warning("Using model under Academic Software License (ASL).")
1012
- selected_default_dtype = st.sidebar.selectbox("Select Precision (default_dtype):", ['float32', 'float64'])
 
1013
  if model_type == "FairChem":
1014
  selected_model = st.sidebar.selectbox("Select FairChem Model:", list(FAIRCHEM_MODELS.keys()))
1015
  model_path = FAIRCHEM_MODELS[selected_model]
@@ -1026,7 +1027,8 @@ if model_type == "ORB":
1026
  model_path = ORB_MODELS[selected_model]
1027
  # if "omat" in selected_model:
1028
  # st.sidebar.warning("Using model under Academic Software License (ASL) license, see [https://github.com/gabor1/ASL](https://github.com/gabor1/ASL). To use this model you accept the terms of the license.")
1029
- selected_default_dtype = st.sidebar.selectbox("Select Precision (default_dtype):", ['float32-high', 'float32-highest', 'float64'])
 
1030
  if model_type == "MatterSim":
1031
  selected_model = st.sidebar.selectbox("Select MatterSim Model:", list(MATTERSIM_MODELS.keys()))
1032
  model_path = MATTERSIM_MODELS[selected_model]
 
1009
  model_path = MACE_MODELS[selected_model]
1010
  if selected_model == "MACE OMAT Medium":
1011
  st.sidebar.warning("Using model under Academic Software License (ASL).")
1012
+ # selected_default_dtype = st.sidebar.selectbox("Select Precision (default_dtype):", ['float32', 'float64'])
1013
+ selected_default_dtype = 'float64'
1014
  if model_type == "FairChem":
1015
  selected_model = st.sidebar.selectbox("Select FairChem Model:", list(FAIRCHEM_MODELS.keys()))
1016
  model_path = FAIRCHEM_MODELS[selected_model]
 
1027
  model_path = ORB_MODELS[selected_model]
1028
  # if "omat" in selected_model:
1029
  # st.sidebar.warning("Using model under Academic Software License (ASL) license, see [https://github.com/gabor1/ASL](https://github.com/gabor1/ASL). To use this model you accept the terms of the license.")
1030
+ # selected_default_dtype = st.sidebar.selectbox("Select Precision (default_dtype):", ['float32-high', 'float32-highest', 'float64'])
1031
+ selected_default_dtype = 'float64'
1032
  if model_type == "MatterSim":
1033
  selected_model = st.sidebar.selectbox("Select MatterSim Model:", list(MATTERSIM_MODELS.keys()))
1034
  model_path = MATTERSIM_MODELS[selected_model]