NGUYEN, Xuan Phi commited on
Commit
c14f353
1 Parent(s): f028d50
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -347,7 +347,8 @@ def llama_load_weights(
347
 
348
 
349
  # Reassign LlamaForCausalLM.load_weights with llama_load_weights
350
- LlamaForCausalLM.load_weights = llama_load_weights
 
351
 
352
  # ! ==================================================================
353
 
 
347
 
348
 
349
  # Reassign LlamaForCausalLM.load_weights with llama_load_weights
350
+ if not DEBUG:
351
+ LlamaForCausalLM.load_weights = llama_load_weights
352
 
353
  # ! ==================================================================
354