linjc16 commited on
Commit
136fe04
·
verified ·
1 Parent(s): ac7bbf3

Update README.md

Browse files

Load the model in the following way (same as Mistral):

```[python]
from transformers import AutoModelForCausalLM, AutoTokenizer
import torch

model_id = 'linjc16/Panacea-7B-Chat'

model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16, device_map="auto", cache_dir="/data/jl254/cache/")
tokenizer = AutoTokenizer.from_pretrained(model_id)

Files changed (1) hide show
  1. README.md +11 -3
README.md CHANGED
@@ -1,3 +1,11 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ language:
4
+ - en
5
+ base_model:
6
+ - mistralai/Mistral-7B-v0.1
7
+ pipeline_tag: text-generation
8
+ tags:
9
+ - clinical trial
10
+ - foundation model
11
+ ---