Spaces:
Running
Running
Commit
·
627f0ff
1
Parent(s):
70c0578
Update app.py
Browse files
app.py
CHANGED
|
@@ -16,7 +16,7 @@ device = (
|
|
| 16 |
else "cpu"
|
| 17 |
)
|
| 18 |
|
| 19 |
-
pipeline_name = 'WiNE-iNEFF/Minecraft-Skin-Diffusion-
|
| 20 |
image_pipe = DDPMPipeline.from_pretrained(pipeline_name, use_safetensors=True).to(device)
|
| 21 |
try:
|
| 22 |
scheduler = DDIMScheduler.from_pretrained(pipeline_name)
|
|
|
|
| 16 |
else "cpu"
|
| 17 |
)
|
| 18 |
|
| 19 |
+
pipeline_name = 'WiNE-iNEFF/Minecraft-Skin-Diffusion-V2'
|
| 20 |
image_pipe = DDPMPipeline.from_pretrained(pipeline_name, use_safetensors=True).to(device)
|
| 21 |
try:
|
| 22 |
scheduler = DDIMScheduler.from_pretrained(pipeline_name)
|