Update app.py
Browse files
app.py
CHANGED
|
@@ -77,13 +77,13 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
|
| 77 |
#vae=AutoencoderKL.from_pretrained("ford442/sdxl-vae-bf16", use_safetensors=True, subfolder='vae',token=True)
|
| 78 |
#vaeX=AutoencoderKL.from_pretrained("ford442/stable-diffusion-3.5-large-fp32", safety_checker=None, use_safetensors=True, subfolder='vae', low_cpu_mem_usage=False, torch_dtype=torch.float32, token=True)
|
| 79 |
|
| 80 |
-
|
| 81 |
-
|
| 82 |
-
|
| 83 |
-
|
| 84 |
-
|
| 85 |
|
| 86 |
-
pipe = StableDiffusion3Pipeline.from_pretrained(
|
| 87 |
#"stabilityai # stable-diffusion-3.5-large",
|
| 88 |
#"ford442/stable-diffusion-3.5-large-bf16",
|
| 89 |
"ford442/stable-diffusion-3.5-large-fp32",
|
|
@@ -99,10 +99,10 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
|
| 99 |
#tokenizer=CLIPTokenizer.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", add_prefix_space=True, subfolder="tokenizer", token=True),
|
| 100 |
#tokenizer_2=CLIPTokenizer.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", add_prefix_space=True, subfolder="tokenizer_2", token=True),
|
| 101 |
#tokenizer_3=T5TokenizerFast.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", add_prefix_space=False, use_fast=True, subfolder="tokenizer_3", token=True),
|
| 102 |
-
tokenizer_3=T5TokenizerFast.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", add_prefix_space=False, use_fast=True, subfolder="tokenizer_3", token=True),
|
| 103 |
#torch_dtype=torch.bfloat16,
|
| 104 |
#use_safetensors=False,
|
| 105 |
-
)
|
| 106 |
#text_encoder=CLIPTextModelWithProjection.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", subfolder='text_encoder', token=True).to(torch.device("cuda:0")) #, dtype=torch.bfloat16)
|
| 107 |
#text_encoder_2=CLIPTextModelWithProjection.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", subfolder='text_encoder_2',token=True).to(torch.device("cuda:0")) #, dtype=torch.bfloat16)
|
| 108 |
#text_encoder_3=T5EncoderModel.from_pretrained("ford442/stable-diffusion-3.5-large-fp32", subfolder='text_encoder_3',token=True).to(torch.device("cuda:0")) #, dtype=torch.bfloat16)
|
|
|
|
| 77 |
#vae=AutoencoderKL.from_pretrained("ford442/sdxl-vae-bf16", use_safetensors=True, subfolder='vae',token=True)
|
| 78 |
#vaeX=AutoencoderKL.from_pretrained("ford442/stable-diffusion-3.5-large-fp32", safety_checker=None, use_safetensors=True, subfolder='vae', low_cpu_mem_usage=False, torch_dtype=torch.float32, token=True)
|
| 79 |
|
| 80 |
+
pipe = StableDiffusion3Pipeline.from_single_file(
|
| 81 |
+
"https://huggingface.co/1inkus/sd35-large-UltraReal-bf16-DDUF/resolve/main/sd3-bf16-large.dduf",
|
| 82 |
+
tokenizer_3=T5TokenizerFast.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", add_prefix_space=False, use_fast=True, subfolder="tokenizer_3"),
|
| 83 |
+
use_safetensors=True,
|
| 84 |
+
)
|
| 85 |
|
| 86 |
+
### pipe = StableDiffusion3Pipeline.from_pretrained(
|
| 87 |
#"stabilityai # stable-diffusion-3.5-large",
|
| 88 |
#"ford442/stable-diffusion-3.5-large-bf16",
|
| 89 |
"ford442/stable-diffusion-3.5-large-fp32",
|
|
|
|
| 99 |
#tokenizer=CLIPTokenizer.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", add_prefix_space=True, subfolder="tokenizer", token=True),
|
| 100 |
#tokenizer_2=CLIPTokenizer.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", add_prefix_space=True, subfolder="tokenizer_2", token=True),
|
| 101 |
#tokenizer_3=T5TokenizerFast.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", add_prefix_space=False, use_fast=True, subfolder="tokenizer_3", token=True),
|
| 102 |
+
### tokenizer_3=T5TokenizerFast.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", add_prefix_space=False, use_fast=True, subfolder="tokenizer_3", token=True),
|
| 103 |
#torch_dtype=torch.bfloat16,
|
| 104 |
#use_safetensors=False,
|
| 105 |
+
###)
|
| 106 |
#text_encoder=CLIPTextModelWithProjection.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", subfolder='text_encoder', token=True).to(torch.device("cuda:0")) #, dtype=torch.bfloat16)
|
| 107 |
#text_encoder_2=CLIPTextModelWithProjection.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", subfolder='text_encoder_2',token=True).to(torch.device("cuda:0")) #, dtype=torch.bfloat16)
|
| 108 |
#text_encoder_3=T5EncoderModel.from_pretrained("ford442/stable-diffusion-3.5-large-fp32", subfolder='text_encoder_3',token=True).to(torch.device("cuda:0")) #, dtype=torch.bfloat16)
|