santafixer-demo

Sleeping

codelion commited on Apr 14, 2023

Commit

9453e0d

1 Parent(s): 54683ea

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -28,8 +28,8 @@ tokenizer_fim.add_special_tokens({
   "pad_token": EOD,
 })
-tokenizer = AutoTokenizer.from_pretrained("bigcode/christmas-models", use_auth_token=token)
-model = AutoModelForCausalLM.from_pretrained("bigcode/christmas-models", trust_remote_code=True, use_auth_token=token).to(device)
 pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, device=device)
 def post_processing(prompt, completion):

   "pad_token": EOD,
 })
+tokenizer = AutoTokenizer.from_pretrained("bigcode/christmas-models")
+model = AutoModelForCausalLM.from_pretrained("bigcode/christmas-models", trust_remote_code=True).to(device)
 pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, device=device)
 def post_processing(prompt, completion):