Update README.md
Browse files
README.md
CHANGED
|
@@ -199,6 +199,7 @@ model = OvisLlamaGPTQForCausalLM.from_pretrained(
|
|
| 199 |
llm_attn_implementation='eager',
|
| 200 |
trust_remote_code=True
|
| 201 |
).cuda()
|
|
|
|
| 202 |
print(f"Model Loaded!")
|
| 203 |
|
| 204 |
|
|
|
|
| 199 |
llm_attn_implementation='eager',
|
| 200 |
trust_remote_code=True
|
| 201 |
).cuda()
|
| 202 |
+
model.model.llm.model.config.use_cache = False
|
| 203 |
print(f"Model Loaded!")
|
| 204 |
|
| 205 |
|