TryingHard
commited on
Commit
•
da09c52
1
Parent(s):
d034c7a
Update README.md
Browse files
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.
|
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
|