Hugging Face's logo

Spaces:
netgvarun2005
/
VirtualTherapist
Sleeping

App Files Community
sharmavaruncs commited on
Commit
5b8a196
·
1 Parent(s): b804250

added cuda()

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -218,7 +218,7 @@ def load_model():
218
  #tokenizer_gpt = AutoTokenizer.from_pretrained("netgvarun2005/GPTVirtualTherapistTokenizer", pad_token='<|pad|>',bos_token='<|startoftext|>',eos_token='<|endoftext|>')
219
  tokenizer_gpt = AutoTokenizer.from_pretrained("netgvarun2005/GPTTherapistDeepSpeedTokenizer", pad_token='<|pad|>',bos_token='<|startoftext|>',eos_token='<|endoftext|>')
220
  #model_gpt = AutoModelForCausalLM.from_pretrained("netgvarun2005/GPTVirtualTherapist")
221
- model_gpt = AutoModelForCausalLM.from_pretrained("netgvarun2005/GPTTherapistDeepSpeedModel")
222
 
223
  return multiModel,tokenizer,model_gpt,tokenizer_gpt
224
 
 
218
  #tokenizer_gpt = AutoTokenizer.from_pretrained("netgvarun2005/GPTVirtualTherapistTokenizer", pad_token='<|pad|>',bos_token='<|startoftext|>',eos_token='<|endoftext|>')
219
  tokenizer_gpt = AutoTokenizer.from_pretrained("netgvarun2005/GPTTherapistDeepSpeedTokenizer", pad_token='<|pad|>',bos_token='<|startoftext|>',eos_token='<|endoftext|>')
220
  #model_gpt = AutoModelForCausalLM.from_pretrained("netgvarun2005/GPTVirtualTherapist")
221
+ model_gpt = AutoModelForCausalLM.from_pretrained("netgvarun2005/GPTTherapistDeepSpeedModel").cuda()
222
 
223
  return multiModel,tokenizer,model_gpt,tokenizer_gpt
224