zw89 commited on
Commit
631bc3a
·
verified ·
1 Parent(s): 4c73558

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,7 +16,7 @@ pipe = StableDiffusionPipeline.from_single_file(model_path, torch_dtype=torch.fl
16
  pipe.enable_attention_slicing()
17
 
18
  # 2. Ускорители
19
- model_id = "Lykon/dreamshaper-8-lcm" # Используем оптимизированную версию
20
  pipe = StableDiffusionPipeline.from_pretrained(
21
  model_id,
22
  torch_dtype=torch.float32,
@@ -30,7 +30,7 @@ def generate(prompt, negative_prompt):
30
 
31
  image = pipe(
32
  prompt=prompt,
33
- negative_prompt=negative_prompt, # Теперь берется из интерфейса
34
  num_inference_steps=6,
35
  guidance_scale=1.5,
36
  width=512,
 
16
  pipe.enable_attention_slicing()
17
 
18
  # 2. Ускорители
19
+ model_id = "zw89/dreamshaper-8-lcm" # Используем оптимизированную версию
20
  pipe = StableDiffusionPipeline.from_pretrained(
21
  model_id,
22
  torch_dtype=torch.float32,
 
30
 
31
  image = pipe(
32
  prompt=prompt,
33
+ negative_prompt=negative_prompt,
34
  num_inference_steps=6,
35
  guidance_scale=1.5,
36
  width=512,