Spaces:
Sleeping
Sleeping
Commit
·
ae9568a
1
Parent(s):
150947d
fix for gemma
Browse files- bp_phi/llm_iface.py +1 -1
bp_phi/llm_iface.py
CHANGED
|
@@ -5,7 +5,7 @@ import torch, random, numpy as np
|
|
| 5 |
from transformers import AutoModelForCausalLM, AutoTokenizer, set_seed
|
| 6 |
from typing import List, Optional
|
| 7 |
|
| 8 |
-
DEBUG =
|
| 9 |
|
| 10 |
def dbg(*args):
|
| 11 |
if DEBUG:
|
|
|
|
| 5 |
from transformers import AutoModelForCausalLM, AutoTokenizer, set_seed
|
| 6 |
from typing import List, Optional
|
| 7 |
|
| 8 |
+
DEBUG = 1
|
| 9 |
|
| 10 |
def dbg(*args):
|
| 11 |
if DEBUG:
|