Hugging Face's logo

Spaces:
tomg-group-umd
/
lm-watermarking
Runtime error

App Files Community
1
jwkirchenbauer commited on
Commit
3682749
•
1 Parent(s): fa85da4

commented bloom/z models due to bug?

Browse files
Files changed (1) hide show
  1. demo_watermark.py +2 -2
demo_watermark.py CHANGED
@@ -43,11 +43,11 @@ from watermark_processor import WatermarkLogitsProcessor, WatermarkDetector
43
 
44
  # FIXME correct lengths for all models
45
  API_MODEL_MAP = {
46
- "bigscience/bloom" : {"max_length": 1000, "gamma": 0.5, "delta": 2.0},
47
- "bigscience/bloomz" : {"max_length": 1000, "gamma": 0.5, "delta": 2.0},
48
  "google/flan-ul2" : {"max_length": 1000, "gamma": 0.5, "delta": 2.0},
49
  "google/flan-t5-xxl" : {"max_length": 1000, "gamma": 0.5, "delta": 2.0},
50
  "EleutherAI/gpt-neox-20b" : {"max_length": 1000, "gamma": 0.5, "delta": 2.0},
 
 
51
  }
52
 
53
  def str2bool(v):
 
43
 
44
  # FIXME correct lengths for all models
45
  API_MODEL_MAP = {
 
 
46
  "google/flan-ul2" : {"max_length": 1000, "gamma": 0.5, "delta": 2.0},
47
  "google/flan-t5-xxl" : {"max_length": 1000, "gamma": 0.5, "delta": 2.0},
48
  "EleutherAI/gpt-neox-20b" : {"max_length": 1000, "gamma": 0.5, "delta": 2.0},
49
+ # "bigscience/bloom" : {"max_length": 1000, "gamma": 0.5, "delta": 2.0},
50
+ # "bigscience/bloomz" : {"max_length": 1000, "gamma": 0.5, "delta": 2.0},
51
  }
52
 
53
  def str2bool(v):