Spaces:

wangrongsheng
/

CareLlama

Runtime error

wangrongsheng commited on Sep 7, 2023

Commit

625f8fd

•

1 Parent(s): 36a5068

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -4,7 +4,7 @@ from typing import Iterator
 import torch
 from transformers import AutoConfig, AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
-model_id = 'wangrongsheng/CareLlama2-7b-super-mix'
 if torch.cuda.is_available():
     config = AutoConfig.from_pretrained(model_id)

 import torch
 from transformers import AutoConfig, AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
+model_id = 'wangrongsheng/CareLlama2-7b-multi'
 if torch.cuda.is_available():
     config = AutoConfig.from_pretrained(model_id)