Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -31,7 +31,7 @@ torch.backends.cudnn.deterministic = False
|
|
31 |
torch.backends.cudnn.benchmark = False
|
32 |
torch.set_float32_matmul_precision("highest")
|
33 |
|
34 |
-
from diffusers import StableDiffusion3Pipeline, SD3Transformer2DModel, AutoencoderKL
|
35 |
from PIL import Image
|
36 |
from image_gen_aux import UpscaleWithModel
|
37 |
|
@@ -73,16 +73,12 @@ pipe = StableDiffusion3Pipeline.from_pretrained(
|
|
73 |
"ford442/stable-diffusion-3.5-large-bf16",
|
74 |
trust_remote_code=True,
|
75 |
transformer=None, # Load transformer separately
|
76 |
-
#scheduler=None,
|
77 |
use_safetensors=True
|
78 |
)
|
79 |
|
80 |
-
#scheduler2=FlowMatchEulerDiscreteScheduler.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", subfolder='scheduler')
|
81 |
-
#pipe.scheduler=scheduler2
|
82 |
ll_transformer=SD3Transformer2DModel.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", subfolder='transformer').to(device, dtype=torch.bfloat16)
|
83 |
pipe.transformer=ll_transformer
|
84 |
pipe.load_lora_weights("ford442/sdxl-vae-bf16", weight_name="LoRA/UltraReal.safetensors")
|
85 |
-
|
86 |
pipe.to(device=device, dtype=torch.bfloat16)
|
87 |
|
88 |
upscaler_2 = UpscaleWithModel.from_pretrained("Kim2091/ClearRealityV1").to(device)
|
|
|
31 |
torch.backends.cudnn.benchmark = False
|
32 |
torch.set_float32_matmul_precision("highest")
|
33 |
|
34 |
+
from diffusers import StableDiffusion3Pipeline, SD3Transformer2DModel, AutoencoderKL
|
35 |
from PIL import Image
|
36 |
from image_gen_aux import UpscaleWithModel
|
37 |
|
|
|
73 |
"ford442/stable-diffusion-3.5-large-bf16",
|
74 |
trust_remote_code=True,
|
75 |
transformer=None, # Load transformer separately
|
|
|
76 |
use_safetensors=True
|
77 |
)
|
78 |
|
|
|
|
|
79 |
ll_transformer=SD3Transformer2DModel.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", subfolder='transformer').to(device, dtype=torch.bfloat16)
|
80 |
pipe.transformer=ll_transformer
|
81 |
pipe.load_lora_weights("ford442/sdxl-vae-bf16", weight_name="LoRA/UltraReal.safetensors")
|
|
|
82 |
pipe.to(device=device, dtype=torch.bfloat16)
|
83 |
|
84 |
upscaler_2 = UpscaleWithModel.from_pretrained("Kim2091/ClearRealityV1").to(device)
|