Spaces:
Running
on
Zero
Running
on
Zero
prithivMLmods
commited on
Commit
•
8d2d89a
1
Parent(s):
04bdf31
Update app.py
Browse files
app.py
CHANGED
@@ -13,7 +13,6 @@ import os
|
|
13 |
import random
|
14 |
import uuid
|
15 |
from typing import Tuple
|
16 |
-
|
17 |
import gradio as gr
|
18 |
import numpy as np
|
19 |
from PIL import Image
|
@@ -23,9 +22,7 @@ from diffusers import StableDiffusionXLPipeline, EulerAncestralDiscreteScheduler
|
|
23 |
|
24 |
DESCRIPTIONz= """## EPIC REALISM 🙀
|
25 |
|
26 |
-
|
27 |
"""
|
28 |
-
|
29 |
def save_image(img):
|
30 |
unique_name = str(uuid.uuid4()) + ".png"
|
31 |
img.save(unique_name)
|
@@ -53,7 +50,7 @@ if torch.cuda.is_available():
|
|
53 |
use_safetensors=True,
|
54 |
)
|
55 |
pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
|
56 |
-
pipe.load_lora_weights("prithivMLmods/
|
57 |
pipe.set_adapters("rlms")
|
58 |
pipe.to("cuda")
|
59 |
|
|
|
13 |
import random
|
14 |
import uuid
|
15 |
from typing import Tuple
|
|
|
16 |
import gradio as gr
|
17 |
import numpy as np
|
18 |
from PIL import Image
|
|
|
22 |
|
23 |
DESCRIPTIONz= """## EPIC REALISM 🙀
|
24 |
|
|
|
25 |
"""
|
|
|
26 |
def save_image(img):
|
27 |
unique_name = str(uuid.uuid4()) + ".png"
|
28 |
img.save(unique_name)
|
|
|
50 |
use_safetensors=True,
|
51 |
)
|
52 |
pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
|
53 |
+
pipe.load_lora_weights("prithivMLmods/Canopus-Realism-LoRA", weight_name="Canopus-Realism-LoRA.safetensors", adapter_name="rlms")
|
54 |
pipe.set_adapters("rlms")
|
55 |
pipe.to("cuda")
|
56 |
|