Spaces:
Paused
Paused
Disabled GPU
Browse files
app.py
CHANGED
|
@@ -7,10 +7,10 @@ from diffusers import StableDiffusionInpaintPipeline
|
|
| 7 |
|
| 8 |
pipeline = StableDiffusionInpaintPipeline.from_pretrained("runwayml/stable-diffusion-inpainting",
|
| 9 |
revision="fp16",
|
| 10 |
-
torch_dtype=torch.float16,
|
| 11 |
safety_checker=lambda images, **kwargs: (images, False))
|
| 12 |
|
| 13 |
-
pipeline.to("cuda")
|
| 14 |
#generator = torch.Generator(device).manual_seed(seed)
|
| 15 |
|
| 16 |
def diffuse(prompt, negativePrompt, inputImage, mask, guidanceScale, numInferenceSteps):
|
|
|
|
| 7 |
|
| 8 |
pipeline = StableDiffusionInpaintPipeline.from_pretrained("runwayml/stable-diffusion-inpainting",
|
| 9 |
revision="fp16",
|
| 10 |
+
#torch_dtype=torch.float16,
|
| 11 |
safety_checker=lambda images, **kwargs: (images, False))
|
| 12 |
|
| 13 |
+
#pipeline.to("cuda")
|
| 14 |
#generator = torch.Generator(device).manual_seed(seed)
|
| 15 |
|
| 16 |
def diffuse(prompt, negativePrompt, inputImage, mask, guidanceScale, numInferenceSteps):
|