manbeast3b commited on
Commit
251c9b7
·
verified ·
1 Parent(s): 0636951

Update src/pipeline.py

Browse files
Files changed (1) hide show
  1. src/pipeline.py +1 -0
src/pipeline.py CHANGED
@@ -17,6 +17,7 @@ torch.backends.cudnn.benchmark = True
17
  torch.backends.cuda.matmul.allow_tf32 = True
18
  torch.cuda.set_per_process_memory_fraction(0.999)
19
 
 
20
  class BasicQuantization:
21
  def __init__(self, bits=1):
22
  self.bits = bits
 
17
  torch.backends.cuda.matmul.allow_tf32 = True
18
  torch.cuda.set_per_process_memory_fraction(0.999)
19
 
20
+
21
  class BasicQuantization:
22
  def __init__(self, bits=1):
23
  self.bits = bits