Spaces:

KennethTM
/

scientific_writing_assistant

Running

KennethTM commited on Apr 26, 2023

Commit

d35712c

•

1 Parent(s): 7cf951b

Set num_gpus=0

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ import galai as gal
 #@st.cache(suppress_st_warning=True, allow_output_mutation=True)
 @st.cache_resource
 def load_model(model_name):
-    model = gal.load_model(model_name) #, dtype=torch.float16
     return model
 if 'text' not in st.session_state:

 #@st.cache(suppress_st_warning=True, allow_output_mutation=True)
 @st.cache_resource
 def load_model(model_name):
+    model = gal.load_model(model_name, num_gpus=0) #, dtype=torch.float16
     return model
 if 'text' not in st.session_state: