qingshan777 commited on
Commit
102ae2e
1 Parent(s): 5e09d5a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -64,7 +64,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
64
  import torch
65
  from PIL import Image
66
 
67
- checkpoint = "qh360_vl-8B"
68
 
69
  model = AutoModelForCausalLM.from_pretrained(checkpoint, torch_dtype=torch.float16, device_map='cuda', trust_remote_code=True).eval()
70
  tokenizer = AutoTokenizer.from_pretrained(checkpoint, trust_remote_code=True)
 
64
  import torch
65
  from PIL import Image
66
 
67
+ checkpoint = "qihoo360/360VL-8B"
68
 
69
  model = AutoModelForCausalLM.from_pretrained(checkpoint, torch_dtype=torch.float16, device_map='cuda', trust_remote_code=True).eval()
70
  tokenizer = AutoTokenizer.from_pretrained(checkpoint, trust_remote_code=True)