Spaces:
Running
on
Zero
Running
on
Zero
NGUYEN, Xuan Phi
commited on
Commit
•
c14f353
1
Parent(s):
f028d50
ujpdate
Browse files
app.py
CHANGED
@@ -347,7 +347,8 @@ def llama_load_weights(
|
|
347 |
|
348 |
|
349 |
# Reassign LlamaForCausalLM.load_weights with llama_load_weights
|
350 |
-
|
|
|
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 |
|