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")
|