Yaron Koresh commited on
Commit
2b8a9a3
·
verified ·
1 Parent(s): 87a02d8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -30,7 +30,7 @@ from huggingface_hub import hf_hub_download
30
  from safetensors.torch import load_file, save_file
31
  from diffusers import FluxPipeline
32
  from PIL import Image, ImageDraw, ImageFont
33
- from transformers import pipeline, T5ForConditionalGeneration, T5Tokenizer
34
  from refiners.fluxion.utils import manual_seed
35
  from refiners.foundationals.latent_diffusion import Solver, solvers
36
  from refiners.foundationals.latent_diffusion.stable_diffusion_1.multi_upscaler import (
@@ -39,7 +39,7 @@ from refiners.foundationals.latent_diffusion.stable_diffusion_1.multi_upscaler i
39
  )
40
  from datetime import datetime
41
 
42
- model = T5ForConditionalGeneration.from_pretrained("google/mt5-xl")
43
  tokenizer = T5Tokenizer.from_pretrained("google/mt5-xl")
44
 
45
  def log(msg):
 
30
  from safetensors.torch import load_file, save_file
31
  from diffusers import FluxPipeline
32
  from PIL import Image, ImageDraw, ImageFont
33
+ from transformers import pipeline, MT5ForConditionalGeneration, T5Tokenizer
34
  from refiners.fluxion.utils import manual_seed
35
  from refiners.foundationals.latent_diffusion import Solver, solvers
36
  from refiners.foundationals.latent_diffusion.stable_diffusion_1.multi_upscaler import (
 
39
  )
40
  from datetime import datetime
41
 
42
+ model = MT5ForConditionalGeneration.from_pretrained("google/mt5-xl")
43
  tokenizer = T5Tokenizer.from_pretrained("google/mt5-xl")
44
 
45
  def log(msg):