Update app.py
Browse files
app.py
CHANGED
|
@@ -7,18 +7,10 @@ from multit2i import (load_models, infer_fn, infer_rand_fn, save_gallery,
|
|
| 7 |
import os
|
| 8 |
os.system("pip install --upgrade diffusers")
|
| 9 |
|
| 10 |
-
from diffusers import StableDiffusionPipeline
|
| 11 |
import torch
|
| 12 |
|
| 13 |
-
# Указываем пути к модели и LoRA
|
| 14 |
-
model_path = "black-forest-labs/FLUX.1-dev"
|
| 15 |
-
lora_path = "MrDrmm/Test_only" # Например: "./models/lora/my_lora.safetensors"
|
| 16 |
|
| 17 |
-
# Загружаем основную модель
|
| 18 |
-
pipe = StableDiffusionPipeline.from_pretrained(model_path, torch_dtype=torch.float32)
|
| 19 |
-
|
| 20 |
-
# Загружаем LoRA
|
| 21 |
-
pipe.load_lora_weights(lora_path)
|
| 22 |
|
| 23 |
# Переключаем на CPU
|
| 24 |
pipe.to("cpu")
|
|
|
|
| 7 |
import os
|
| 8 |
os.system("pip install --upgrade diffusers")
|
| 9 |
|
| 10 |
+
from diffusers import StableDiffusionPipeline
|
| 11 |
import torch
|
| 12 |
|
|
|
|
|
|
|
|
|
|
| 13 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 14 |
|
| 15 |
# Переключаем на CPU
|
| 16 |
pipe.to("cpu")
|