James Peter Perrfone Jefferies
commited on
Commit
·
798391a
1
Parent(s):
601904e
Re-enable the safety checker
Browse files
app.py
CHANGED
|
@@ -7,7 +7,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
| 7 |
def generate(
|
| 8 |
prompt, negative_prompt, num_inference_steps, width, height, guidance_scale, seed
|
| 9 |
):
|
| 10 |
-
pipeline = DiffusionPipeline.from_pretrained("Lykon/DreamShaper"
|
| 11 |
pipeline = pipeline.to(device)
|
| 12 |
generator = torch.Generator(device=device).manual_seed(seed)
|
| 13 |
|
|
|
|
| 7 |
def generate(
|
| 8 |
prompt, negative_prompt, num_inference_steps, width, height, guidance_scale, seed
|
| 9 |
):
|
| 10 |
+
pipeline = DiffusionPipeline.from_pretrained("Lykon/DreamShaper")
|
| 11 |
pipeline = pipeline.to(device)
|
| 12 |
generator = torch.Generator(device=device).manual_seed(seed)
|
| 13 |
|