ford442 commited on
Commit
866c000
·
verified ·
1 Parent(s): d31f3b7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -2,6 +2,8 @@ import spaces
2
  import gradio as gr
3
  import numpy as np
4
  import random
 
 
5
  import torch
6
  from diffusers import StableDiffusion3Pipeline
7
  #from transformers import CLIPTextModelWithProjection, T5EncoderModel
@@ -76,11 +78,13 @@ pipe = StableDiffusion3Pipeline.from_pretrained(
76
  #use_safetensors=False,
77
  )
78
 
 
 
 
 
79
  #pipe.to(device=device, dtype=torch.bfloat16)
80
  pipe.to(device)
81
 
82
- pipe.load_lora_weights("ford442/sdxl-vae-bf16", weight_name="LoRA/UltraReal.safetensors", adapter_name="real")
83
-
84
  upscaler_2 = UpscaleWithModel.from_pretrained("Kim2091/ClearRealityV1").to(torch.device('cpu'))
85
 
86
  MAX_SEED = np.iinfo(np.int32).max
 
2
  import gradio as gr
3
  import numpy as np
4
  import random
5
+ from lora import LoRAModel
6
+
7
  import torch
8
  from diffusers import StableDiffusion3Pipeline
9
  #from transformers import CLIPTextModelWithProjection, T5EncoderModel
 
78
  #use_safetensors=False,
79
  )
80
 
81
+
82
+
83
+ lora_model = LoRAModel.from_pretrained("https://huggingface.co/ford442/sdxl-vae-bf16/LoRA/UltraReal.safetensors")
84
+ pipe.apply_lora(lora_model, scaling_factor=0.75)
85
  #pipe.to(device=device, dtype=torch.bfloat16)
86
  pipe.to(device)
87
 
 
 
88
  upscaler_2 = UpscaleWithModel.from_pretrained("Kim2091/ClearRealityV1").to(torch.device('cpu'))
89
 
90
  MAX_SEED = np.iinfo(np.int32).max