Spaces:
Sleeping
Sleeping
slymnyldrm
commited on
Commit
•
742d313
1
Parent(s):
d484d65
Update app.py
Browse files
app.py
CHANGED
@@ -12,9 +12,10 @@ pipe = StableDiffusionPipeline.from_pretrained(model_path, use_safetensors=True,
|
|
12 |
pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
|
13 |
|
14 |
g_cuda = torch.Generator()
|
|
|
15 |
g_cuda.manual_seed(g_seed)
|
16 |
|
17 |
-
def inference(prompt, negative_prompt, num_samples, height=512, width=512, num_inference_steps=50, guidance_scale=7.5
|
18 |
with torch.inference_mode():
|
19 |
return pipe(
|
20 |
prompt, height=int(height), width=int(width),
|
|
|
12 |
pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
|
13 |
|
14 |
g_cuda = torch.Generator()
|
15 |
+
g_seed=52362
|
16 |
g_cuda.manual_seed(g_seed)
|
17 |
|
18 |
+
def inference(prompt, negative_prompt, num_samples, height=512, width=512, num_inference_steps=50, guidance_scale=7.5):
|
19 |
with torch.inference_mode():
|
20 |
return pipe(
|
21 |
prompt, height=int(height), width=int(width),
|