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):
|