MrDrmm commited on
Commit
4c314b0
·
verified ·
1 Parent(s): 085828a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -9
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, load_lora_weights
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")