TryingHard commited on
Commit
45291da
·
verified ·
1 Parent(s): 7d563ef

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -66,7 +66,7 @@ from auto_gptq.modeling import OvisLlamaGPTQForCausalLM
66
 
67
  # load model
68
  load_device = "cuda:0" # customize load device
69
- model = OvisLlamaGPTQForCausalLM.from_pretrained(
70
  "AIDC-AI/Ovis1.6-Llama3.2-3B-GPTQ-Int4",
71
  device=load_device,
72
  trust_remote_code=True
 
66
 
67
  # load model
68
  load_device = "cuda:0" # customize load device
69
+ model = OvisLlamaGPTQForCausalLM.from_quantized(
70
  "AIDC-AI/Ovis1.6-Llama3.2-3B-GPTQ-Int4",
71
  device=load_device,
72
  trust_remote_code=True