Spaces:
Sleeping
Sleeping
Commit
·
fef2ff0
1
Parent(s):
3497946
Update app.py
Browse files
app.py
CHANGED
@@ -6,7 +6,7 @@ from diffusers import StableDiffusionPipeline, DDIMScheduler
|
|
6 |
|
7 |
move_cache()
|
8 |
|
9 |
-
model_path = "slymnyldrm/
|
10 |
|
11 |
pipe = StableDiffusionPipeline.from_pretrained(model_path, use_safetensors=True, safety_checker=None, torch_dtype=torch.float16)
|
12 |
pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
|
@@ -41,6 +41,6 @@ with gr.Blocks() as demo:
|
|
41 |
with gr.Row():
|
42 |
g_seed = gr.Number(label="Random Number Generator Seed", value=52362)
|
43 |
|
44 |
-
run.click(inference, inputs=[prompt, negative_prompt, num_samples, height, width, num_inference_steps, guidance_scale, g_seed], outputs=gallery)
|
45 |
|
46 |
demo.launch()
|
|
|
6 |
|
7 |
move_cache()
|
8 |
|
9 |
+
model_path = "slymnyldrm/dreambooth_usecase_weights"
|
10 |
|
11 |
pipe = StableDiffusionPipeline.from_pretrained(model_path, use_safetensors=True, safety_checker=None, torch_dtype=torch.float16)
|
12 |
pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
|
|
|
41 |
with gr.Row():
|
42 |
g_seed = gr.Number(label="Random Number Generator Seed", value=52362)
|
43 |
|
44 |
+
run.click(inference, inputs=[prompt, negative_prompt, num_samples, height, width, num_inference_steps, guidance_scale, int(g_seed)], outputs=gallery)
|
45 |
|
46 |
demo.launch()
|