TryingHard commited on
Commit
da09c52
1 Parent(s): d034c7a

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 OvisGemma2GPTQForCausalLM
66
 
67
  # load model
68
  load_device = "cuda:0" # customize load device
69
- model = OvisGemma2GPTQForCausalLM.from_pretrained(
70
  "AIDC-AI/Ovis1.6-Gemma2-9B-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 = OvisGemma2GPTQForCausalLM.from_quantized(
70
  "AIDC-AI/Ovis1.6-Gemma2-9B-GPTQ-Int4",
71
  device=load_device,
72
  trust_remote_code=True