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
|