mobenta commited on
Commit
8be94cc
·
verified ·
1 Parent(s): 29f52f6

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -22,7 +22,7 @@ from PIL import Image
22
  from transformers import AutoModelForCausalLM
23
 
24
  # load model
25
- model = AutoModelForCausalLM.from_pretrained("AIDC-AI/Ovis1.6-Gemma2-9B",
26
  torch_dtype=torch.bfloat16,
27
  multimodal_max_length=8192,
28
  trust_remote_code=True).cuda()
 
22
  from transformers import AutoModelForCausalLM
23
 
24
  # load model
25
+ model = AutoModelForCausalLM.from_pretrained("mobenta/Gemma2-9B",
26
  torch_dtype=torch.bfloat16,
27
  multimodal_max_length=8192,
28
  trust_remote_code=True).cuda()