Update app.py
Browse files
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 = "
|
| 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,
|