alexkueck commited on
Commit
7ef6710
1 Parent(s): 0c9076b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -32,10 +32,9 @@ base_model = "gpt2-xl" # options: ['gpt2', 'gpt2-medium', 'gpt2-large', 'gpt2-
32
  #tokenizer,model,device = load_tokenizer_and_model(base_model,False)
33
 
34
  ################################
35
- #Alternativ: Model und Tokenizer direkt laden
36
- #login(token=os.environ["HF_ACCESS_TOKEN"])
37
- tokenizer = GPT2Tokenizer.from_pretrained(base_model, use_fast = True, use_auth_token=False)
38
- model = GPT2LMHeadModel.from_pretrained(base_model)
39
 
40
 
41
  ########################################################################
 
32
  #tokenizer,model,device = load_tokenizer_and_model(base_model,False)
33
 
34
  ################################
35
+ #Alternativ: Model und Tokenizer für GPT2
36
+ tokenizer,model,device = load_tokenizer_and_model_gpt2(base_model,False)
37
+
 
38
 
39
 
40
  ########################################################################