Update app.py

app.py

@@ -4,7 +4,21 @@ from multit2i import (load_models, infer_fn, infer_rand_fn, save_gallery,
     change_model, warm_model, get_model_info_md, loaded_models,
     get_positive_prefix, get_positive_suffix, get_negative_prefix, get_negative_suffix,
     get_recom_prompt_type, set_recom_prompt_preset, get_tag_type, randomize_seed, translate_to_en)
+from diffusers import StableDiffusionPipeline, load_lora_weights
+import torch
 
+# Указываем пути к модели и LoRA
+model_path = "black-forest-labs/FLUX.1-dev"
+lora_path = "MrDrmm/Test_only"  # Например: "./models/lora/my_lora.safetensors"
+
+# Загружаем основную модель
+pipe = StableDiffusionPipeline.from_pretrained(model_path, torch_dtype=torch.float32)
+
+# Загружаем LoRA
+pipe.load_lora_weights(lora_path)
+
+# Переключаем на CPU
+pipe.to("cpu")
 
 max_images = 8
 MAX_SEED = 2**32-1