lvwerra HF staff commited on
Commit
4d1f3e5
1 Parent(s): f0a40b9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ example = [
13
 
14
  token = os.environ["HUB_TOKEN"]
15
  device="cuda:0"
16
- revision = "dedup-alt"
17
 
18
  tokenizer = AutoTokenizer.from_pretrained("bigcode/christmas-models", use_auth_token=token)
19
  model = AutoModelForCausalLM.from_pretrained("bigcode/christmas-models", revision=revision, trust_remote_code=True, use_auth_token=token)
 
13
 
14
  token = os.environ["HUB_TOKEN"]
15
  device="cuda:0"
16
+ revision = "dedup-alt-comments"
17
 
18
  tokenizer = AutoTokenizer.from_pretrained("bigcode/christmas-models", use_auth_token=token)
19
  model = AutoModelForCausalLM.from_pretrained("bigcode/christmas-models", revision=revision, trust_remote_code=True, use_auth_token=token)