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

Revert "change flux model to lite one"

Browse files

This reverts commit 694332de1933f8914e8412201b7b9dd2dfaa47f3.

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