qingshan777 commited on
Commit
dd4f242
1 Parent(s): 924ecfd

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -66,7 +66,7 @@ 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)
71
  vision_tower = model.get_vision_tower()
72
  vision_tower.load_model()
 
66
 
67
  checkpoint = "qihoo360/360VL-8B"
68
 
69
+ model = AutoModelForCausalLM.from_pretrained(checkpoint, torch_dtype=torch.float16, device_map='auto', trust_remote_code=True).eval()
70
  tokenizer = AutoTokenizer.from_pretrained(checkpoint, trust_remote_code=True)
71
  vision_tower = model.get_vision_tower()
72
  vision_tower.load_model()