vilarin commited on
Commit
3bb671a
·
verified ·
1 Parent(s): 5d1149a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -8
app.py CHANGED
@@ -9,7 +9,7 @@ from diffusers.utils import load_image
9
  from PIL import Image
10
  import requests
11
  import transformers
12
- from transformers import AutoConfig, AutoTokenizer, T5EncoderModel
13
  from translatepy import Translator
14
 
15
  os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
@@ -55,15 +55,9 @@ text_encoder_3 = T5EncoderModel.from_pretrained(
55
  torch_dtype=torch.float16,
56
  )
57
 
58
- model_config = AutoConfig.from_pretrained(
59
  repo,
60
  subfolder="tokenizer_3",
61
- )
62
-
63
- tokenizer_3 = AutoTokenizer.from_pretrained(
64
- repo,
65
- subfolder="tokenizer_3",
66
- config=model_config,
67
  torch_dtype=torch.float16,
68
  )
69
 
 
9
  from PIL import Image
10
  import requests
11
  import transformers
12
+ from transformers import T5Tokenizer, T5EncoderModel
13
  from translatepy import Translator
14
 
15
  os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
 
55
  torch_dtype=torch.float16,
56
  )
57
 
58
+ tokenizer_3 = T5Tokenizer.from_pretrained(
59
  repo,
60
  subfolder="tokenizer_3",
 
 
 
 
 
 
61
  torch_dtype=torch.float16,
62
  )
63