Update educhat_gradio.py

educhat_gradio.py

@@ -63,7 +63,7 @@ def get_specific_model(
     # for now, we can use an argument but in the future,
     # we can automate this
 
-    model = transformers.LlamaForCausalLM.from_pretrained(model_name,torch_dtype=torch.float16, ).half().cuda()
+    model = transformers.LlamaForCausalLM.from_pretrained(model_name,torch_dtype=torch.float16, ).half()
     
     return model