TryingHard commited on
Commit
d29d9f7
1 Parent(s): 45291da

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -0
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