Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -3,7 +3,9 @@ import gradio as gr
|
|
| 3 |
import torch
|
| 4 |
import modin.pandas as pd
|
| 5 |
import numpy as np
|
| 6 |
-
from diffusers import DiffusionPipeline, DPMSolverSinglestepScheduler
|
|
|
|
|
|
|
| 7 |
|
| 8 |
pipe = DiffusionPipeline.from_pretrained("mann-e/Mann-E_Dreams", torch_dtype=torch.float16).to("cuda")
|
| 9 |
pipe.scheduler = DPMSolverSinglestepScheduler.from_config(pipe.scheduler.config, use_karras_sigmas=True)
|
|
@@ -21,7 +23,7 @@ gr.Interface(fn=genie, inputs=[gr.Textbox(label='What you want the AI to generat
|
|
| 21 |
gr.Slider(576, maximum=1280, value=768, step=16, label='Width (can go up to 1280, but for square images maximum is 1024x1024)'),
|
| 22 |
gr.Slider(576, maximum=1280, value=768, step=16, label='Height (can go up to 1280, but for square images maximum is 1024x1024)'),
|
| 23 |
gr.Slider(1, maximum=8, value=6, step=1, label='Number of Iterations'),
|
| 24 |
-
gr.Slider(minimum=0, step=1, maximum=999999999999999999, randomize=True),
|
| 25 |
],
|
| 26 |
outputs='image',
|
| 27 |
title="Mann-E Dreams",
|
|
|
|
| 3 |
import torch
|
| 4 |
import modin.pandas as pd
|
| 5 |
import numpy as np
|
| 6 |
+
from diffusers import DiffusionPipeline, DPMSolverSinglestepScheduler, ControlNetModel, StableDiffusionXLControlNetPipeline, AutoencoderKL
|
| 7 |
+
import requests
|
| 8 |
+
import cv2
|
| 9 |
|
| 10 |
pipe = DiffusionPipeline.from_pretrained("mann-e/Mann-E_Dreams", torch_dtype=torch.float16).to("cuda")
|
| 11 |
pipe.scheduler = DPMSolverSinglestepScheduler.from_config(pipe.scheduler.config, use_karras_sigmas=True)
|
|
|
|
| 23 |
gr.Slider(576, maximum=1280, value=768, step=16, label='Width (can go up to 1280, but for square images maximum is 1024x1024)'),
|
| 24 |
gr.Slider(576, maximum=1280, value=768, step=16, label='Height (can go up to 1280, but for square images maximum is 1024x1024)'),
|
| 25 |
gr.Slider(1, maximum=8, value=6, step=1, label='Number of Iterations'),
|
| 26 |
+
gr.Slider(minimum=0, step=1, maximum=999999999999999999, randomize=True, label="Seed"),
|
| 27 |
],
|
| 28 |
outputs='image',
|
| 29 |
title="Mann-E Dreams",
|