ford442 commited on
Commit
1723a69
·
verified ·
1 Parent(s): 8d68b84

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -6,7 +6,7 @@ import torch
6
  from diffusers import StableDiffusion3Pipeline
7
  #from transformers import CLIPTextModelWithProjection, T5EncoderModel
8
  from transformers import CLIPTokenizer
9
- CLIPTokenizer
10
  import re
11
  import paramiko
12
  import urllib
@@ -65,7 +65,8 @@ pipe = StableDiffusion3Pipeline.from_pretrained(
65
  # text_encoder_2=CLIPTextModelWithProjection.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", subfolder='text_encoder_2',token=True),
66
  # text_encoder_3=T5EncoderModel.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", subfolder='text_encoder_3',token=True),
67
  tokenizer=CLIPTokenizer.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", add_prefix_space=True, subfolder="tokenizer", token=True),
68
- token=True,
 
69
  torch_dtype=torch.bfloat16,
70
  #use_safetensors=False,
71
  )
 
6
  from diffusers import StableDiffusion3Pipeline
7
  #from transformers import CLIPTextModelWithProjection, T5EncoderModel
8
  from transformers import CLIPTokenizer
9
+
10
  import re
11
  import paramiko
12
  import urllib
 
65
  # text_encoder_2=CLIPTextModelWithProjection.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", subfolder='text_encoder_2',token=True),
66
  # text_encoder_3=T5EncoderModel.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", subfolder='text_encoder_3',token=True),
67
  tokenizer=CLIPTokenizer.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", add_prefix_space=True, subfolder="tokenizer", token=True),
68
+ tokenizer_2=CLIPTokenizer.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", add_prefix_space=True, subfolder="tokenizer_2", token=True),
69
+ tokenizer_3=T5TokenizerFast.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", add_prefix_space=True, subfolder="tokenizer_3", token=True),
70
  torch_dtype=torch.bfloat16,
71
  #use_safetensors=False,
72
  )