Spaces:
Running
on
Zero
Running
on
Zero
Upload 3 files
Browse files- README.md +1 -1
- genimage.py +1 -1
- requirements.txt +1 -1
README.md
CHANGED
@@ -4,7 +4,7 @@ emoji: ππ»
|
|
4 |
colorFrom: red
|
5 |
colorTo: purple
|
6 |
sdk: gradio
|
7 |
-
sdk_version: 4.
|
8 |
app_file: app.py
|
9 |
pinned: true
|
10 |
license: apache-2.0
|
|
|
4 |
colorFrom: red
|
5 |
colorTo: purple
|
6 |
sdk: gradio
|
7 |
+
sdk_version: 4.39.0
|
8 |
app_file: app.py
|
9 |
pinned: true
|
10 |
license: apache-2.0
|
genimage.py
CHANGED
@@ -7,7 +7,7 @@ def load_pipeline():
|
|
7 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
8 |
pipe = AutoPipelineForText2Image.from_pretrained(
|
9 |
"John6666/rae-diffusion-xl-v2-sdxl-spo-pcm",
|
10 |
-
custom_pipeline="
|
11 |
torch_dtype=torch.float16,
|
12 |
)
|
13 |
pipe.to(device)
|
|
|
7 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
8 |
pipe = AutoPipelineForText2Image.from_pretrained(
|
9 |
"John6666/rae-diffusion-xl-v2-sdxl-spo-pcm",
|
10 |
+
custom_pipeline="lpw_stable_diffusion_xl",
|
11 |
torch_dtype=torch.float16,
|
12 |
)
|
13 |
pipe.to(device)
|
requirements.txt
CHANGED
@@ -13,5 +13,5 @@ dartrs
|
|
13 |
httpx==0.13.3
|
14 |
httpcore
|
15 |
googletrans==4.0.0rc1
|
16 |
-
|
17 |
rapidfuzz
|
|
|
13 |
httpx==0.13.3
|
14 |
httpcore
|
15 |
googletrans==4.0.0rc1
|
16 |
+
diffusers
|
17 |
rapidfuzz
|