Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -35,7 +35,7 @@ vae = AutoencoderKL.from_pretrained(
|
|
35 |
).to("cuda")
|
36 |
|
37 |
pipe = StableDiffusionXLFillPipeline.from_pretrained(
|
38 |
-
"
|
39 |
torch_dtype=torch.float16,
|
40 |
vae=vae,
|
41 |
controlnet=model,
|
|
|
35 |
).to("cuda")
|
36 |
|
37 |
pipe = StableDiffusionXLFillPipeline.from_pretrained(
|
38 |
+
"imagepipeline/flux_uncensored_nsfw_v2", # Replace with the new model ID
|
39 |
torch_dtype=torch.float16,
|
40 |
vae=vae,
|
41 |
controlnet=model,
|