izumo commited on
Commit
afef663
1 Parent(s): 7db298a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,11 +7,11 @@ import torch.nn.utils.prune as prune
7
  from torch import nn
8
  import torch
9
  from transformers import T5Tokenizer
10
- from transformers import AutoModelForCausalLM
11
 
12
 
13
  model_name = "rinna/japanese-gpt-1b"
14
- model = AutoModelForCausalLM.from_pretrained(model_name)
15
 
16
  tokenizer = T5Tokenizer.from_pretrained(model_name)
17
 
 
7
  from torch import nn
8
  import torch
9
  from transformers import T5Tokenizer
10
+ from transformers import GPT2LMHeadModel
11
 
12
 
13
  model_name = "rinna/japanese-gpt-1b"
14
+ model = GPT2LMHeadModel.from_pretrained(model_name)
15
 
16
  tokenizer = T5Tokenizer.from_pretrained(model_name)
17