TryingHard
commited on
Commit
•
d29d9f7
1
Parent(s):
45291da
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 |
|