1inkusFace commited on
Commit
adb849f
·
verified ·
1 Parent(s): 0b270d8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -5
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, FlowMatchEulerDiscreteScheduler
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)