Spaces:
Running
on
Zero
Running
on
Zero
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 |
|