Spaces:

open-nlp
/

Chris-lab

Sleeping

kz209 commited on Jul 23

Commit

cc6ba40

•

1 Parent(s): a1fddf9

fix

Files changed (1) hide show

pages/summarization_playground.py CHANGED Viewed

@@ -40,7 +40,7 @@ def generate_answer(sources, model, model_name, prompt):
         gc.collect()
         torch.cuda.empty_cache()
-        model[model_name].gpu()
         __model_on_gpu__ = model_name
     answer = model[model_name].gen(content)

         gc.collect()
         torch.cuda.empty_cache()
+        model[model_name].cuda()
         __model_on_gpu__ = model_name
     answer = model[model_name].gen(content)