multimodalart HF staff commited on
Commit
428d2aa
1 Parent(s): 9d41bd5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -25,7 +25,7 @@ unet.load_state_dict(load_file(hf_hub_download(repo, ckpt)))
25
  pipe_lightning = StableDiffusionXLPipeline.from_pretrained(base,
26
  unet=unet,
27
  vae=vae,
28
- text_encoder_1=pipe_turbo.text_encoder_1,
29
  text_encoder_2=pipe_turbo.text_encoder_2,
30
  tokenizer=pipe_turbo.tokenizer,
31
  tokenizer_2=pipe_turbo.tokenizer_2,
@@ -45,7 +45,7 @@ unet.load_state_dict(load_file(hf_hub_download(repo_name, ckpt_name)))
45
  pipe_hyper = StableDiffusionXLPipeline.from_pretrained(base,
46
  unet=unet,
47
  vae=vae,
48
- text_encoder_1=pipe_turbo.text_encoder_1,
49
  text_encoder_2=pipe_turbo.text_encoder_2,
50
  tokenizer=pipe_turbo.tokenizer,
51
  tokenizer_2=pipe_turbo.tokenizer_2,
 
25
  pipe_lightning = StableDiffusionXLPipeline.from_pretrained(base,
26
  unet=unet,
27
  vae=vae,
28
+ text_encoder=pipe_turbo.text_encoder,
29
  text_encoder_2=pipe_turbo.text_encoder_2,
30
  tokenizer=pipe_turbo.tokenizer,
31
  tokenizer_2=pipe_turbo.tokenizer_2,
 
45
  pipe_hyper = StableDiffusionXLPipeline.from_pretrained(base,
46
  unet=unet,
47
  vae=vae,
48
+ text_encoder=pipe_turbo.text_encoder,
49
  text_encoder_2=pipe_turbo.text_encoder_2,
50
  tokenizer=pipe_turbo.tokenizer,
51
  tokenizer_2=pipe_turbo.tokenizer_2,