THUDM
/

cogvlm-grounding-base-hf

Text Generation

Model card Files Files and versions Community

chenkq commited on Nov 20, 2023

Commit

c0445f6

•

1 Parent(s): 1d3912d

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -6,7 +6,7 @@ from PIL import Image
 from transformers import AutoModelForCausalLM, LlamaTokenizer
 tokenizer = LlamaTokenizer.from_pretrained('lmsys/vicuna-7b-v1.5')
 model = AutoModelForCausalLM.from_pretrained(
-    'THUDM/cogvlm-grounding-generalist-hf',
     torch_dtype=torch.bfloat16,
     low_cpu_mem_usage=True,
     trust_remote_code=True

 from transformers import AutoModelForCausalLM, LlamaTokenizer
 tokenizer = LlamaTokenizer.from_pretrained('lmsys/vicuna-7b-v1.5')
 model = AutoModelForCausalLM.from_pretrained(
+    'THUDM/cogvlm-grounding-base-hf',
     torch_dtype=torch.bfloat16,
     low_cpu_mem_usage=True,
     trust_remote_code=True