Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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 |
-
|
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 |
)
|