Spaces:
Sleeping
Sleeping
Commit
·
839ceac
1
Parent(s):
1729c17
changed model to deepspeed
Browse files
app.py
CHANGED
|
@@ -215,8 +215,10 @@ def load_model():
|
|
| 215 |
tokenizer = AutoTokenizer.from_pretrained("netgvarun2005/MultiModalBertHubertTokenizer")
|
| 216 |
|
| 217 |
# GenAI
|
|
|
|
| 218 |
tokenizer_gpt = AutoTokenizer.from_pretrained("netgvarun2005/GPTVirtualTherapistTokenizer", pad_token='<|pad|>',bos_token='<|startoftext|>',eos_token='<|endoftext|>')
|
| 219 |
-
model_gpt = AutoModelForCausalLM.from_pretrained("netgvarun2005/GPTVirtualTherapist")
|
|
|
|
| 220 |
|
| 221 |
return multiModel,tokenizer,model_gpt,tokenizer_gpt
|
| 222 |
|
|
|
|
| 215 |
tokenizer = AutoTokenizer.from_pretrained("netgvarun2005/MultiModalBertHubertTokenizer")
|
| 216 |
|
| 217 |
# GenAI
|
| 218 |
+
#tokenizer_gpt = AutoTokenizer.from_pretrained("netgvarun2005/GPTVirtualTherapistTokenizer", pad_token='<|pad|>',bos_token='<|startoftext|>',eos_token='<|endoftext|>')
|
| 219 |
tokenizer_gpt = AutoTokenizer.from_pretrained("netgvarun2005/GPTVirtualTherapistTokenizer", 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 |
|