EvgenyKu commited on
Commit
694332d
·
1 Parent(s): d27c884

change flux model to lite one

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -65,12 +65,12 @@ pipe = FluxPipeline.from_pretrained(
65
  )
66
  log(f"Загрузка модели FLUX.1-dev завершена за {time.time() - start_time:.2f} секунд")
67
 
68
- start_time = time.time()
69
- log(f"Загрузка LoRA")
70
- pipe.load_lora_weights("Shakker-Labs/FLUX.1-dev-LoRA-add-details", weight_name="FLUX-dev-lora-add_details.safetensors")
71
- log(f"Загрузка LoRA завершена за {time.time() - start_time:.2f} секунд")
72
-
73
- pipe.fuse_lora(lora_scale=0.5)
74
  pipe.to(device)
75
 
76
  start_time = time.time()
 
65
  )
66
  log(f"Загрузка модели FLUX.1-dev завершена за {time.time() - start_time:.2f} секунд")
67
 
68
+ # start_time = time.time()
69
+ # log(f"Загрузка LoRA")
70
+ # pipe.load_lora_weights("Shakker-Labs/FLUX.1-dev-LoRA-add-details", weight_name="FLUX-dev-lora-add_details.safetensors")
71
+ # log(f"Загрузка LoRA завершена за {time.time() - start_time:.2f} секунд")
72
+ #
73
+ # pipe.fuse_lora(lora_scale=0.5)
74
  pipe.to(device)
75
 
76
  start_time = time.time()