Spaces:
Paused
Paused
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)
|