Hugging Face's logo

Spaces:
bigcode
/
santacoder-demo
Runtime error

App Files Community
11
lvwerra HF staff commited on
Commit
757bca5
•
1 Parent(s): 73f3f89

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -13,9 +13,10 @@ example = [
13
 
14
  token = os.environ["HUB_TOKEN"]
15
  device="cpu" #"cuda:0"
 
16
 
17
  tokenizer = AutoTokenizer.from_pretrained("bigcode/christmas-models", use_auth_token=token)
18
- model = AutoModelForCausalLM.from_pretrained("bigcode/christmas-models", revision="dedeup-alt", trust_remote_code=True, use_auth_token=token)
19
 
20
 
21
  def code_generation(gen_prompt, max_tokens, temperature=0.6, seed=42):
 
13
 
14
  token = os.environ["HUB_TOKEN"]
15
  device="cpu" #"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)
20
 
21
 
22
  def code_generation(gen_prompt, max_tokens, temperature=0.6, seed=42):