Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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
|