neuralworm commited on
Commit
ae9568a
·
1 Parent(s): 150947d

fix for gemma

Browse files
Files changed (1) hide show
  1. 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 = os.getenv("BP_PHI_DEBUG", "0") == "1"
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: