Commit ·
664bf39
1
Parent(s): d361487
Update app.py (#2)
Browse files- Update app.py (8421d81eb63244638233bb4ef2fe88ee053f9c30)
Co-authored-by: Roman <Roman190928@users.noreply.huggingface.co>
app.py
CHANGED
|
@@ -5,7 +5,8 @@ import torch
|
|
| 5 |
pipe = AutoPipelineForImage2Image.from_pretrained(
|
| 6 |
"stabilityai/stable-diffusion-xl-base-1.0",
|
| 7 |
torch_dtype=torch.float16
|
| 8 |
-
).to("
|
|
|
|
| 9 |
#pipe.load_lora_weights("enhanceaiteam/Flux-Uncensored-V2")
|
| 10 |
|
| 11 |
def process(image, prompt, strength):
|
|
|
|
| 5 |
pipe = AutoPipelineForImage2Image.from_pretrained(
|
| 6 |
"stabilityai/stable-diffusion-xl-base-1.0",
|
| 7 |
torch_dtype=torch.float16
|
| 8 |
+
).to("cpu")
|
| 9 |
+
#For now, we have to use cpu because CUDA is not available. This taks 8mins per image btw.
|
| 10 |
#pipe.load_lora_weights("enhanceaiteam/Flux-Uncensored-V2")
|
| 11 |
|
| 12 |
def process(image, prompt, strength):
|