alexkueck commited on
Commit
ddbed22
·
1 Parent(s): 3736c10

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -31,7 +31,8 @@ base_model = "gpt2-xl"
31
  #tokenizer,model,device = load_tokenizer_and_model(base_model,False)
32
 
33
  ################################
34
- #Alternativ: Model und Tokenizer direkt ladentokenizer = AutoTokenizer.from_pretrained(base_model, use_fast = True, use_auth_token=True)
 
35
  tokenizer = GPT2Tokenizer.from_pretrained(base_model, use_fast = True, use_auth_token=True)
36
  model = GPT2LMHeadModel.from_pretrained('gpt2') # options: ['gpt2', 'gpt2-medium', 'gpt2-large', 'gpt2-xl']
37
 
 
31
  #tokenizer,model,device = load_tokenizer_and_model(base_model,False)
32
 
33
  ################################
34
+ #Alternativ: Model und Tokenizer direkt laden
35
+ login(token=os.environ["HF_ACCESS_TOKEN"])
36
  tokenizer = GPT2Tokenizer.from_pretrained(base_model, use_fast = True, use_auth_token=True)
37
  model = GPT2LMHeadModel.from_pretrained('gpt2') # options: ['gpt2', 'gpt2-medium', 'gpt2-large', 'gpt2-xl']
38