Spaces:
Runtime error
Runtime error
Revert "change flux model to lite one"
Browse filesThis reverts commit d27c884b2b4b3cfac13719f63ebb6a92f30511fa.
app.py
CHANGED
|
@@ -55,7 +55,7 @@ start_time = time.time()
|
|
| 55 |
log(f"Загрузка модели FLUX.1-dev")
|
| 56 |
pipe = FluxPipeline.from_pretrained(
|
| 57 |
# pretrained_model_name_or_path = local_path,
|
| 58 |
-
"
|
| 59 |
torch_dtype=torch.bfloat16, # Используем bfloat16 для A100
|
| 60 |
# low_cpu_mem_usage=True, # Экономия памяти
|
| 61 |
# device_map="balanced",
|
|
|
|
| 55 |
log(f"Загрузка модели FLUX.1-dev")
|
| 56 |
pipe = FluxPipeline.from_pretrained(
|
| 57 |
# pretrained_model_name_or_path = local_path,
|
| 58 |
+
"black-forest-labs/FLUX.1-dev",
|
| 59 |
torch_dtype=torch.bfloat16, # Используем bfloat16 для A100
|
| 60 |
# low_cpu_mem_usage=True, # Экономия памяти
|
| 61 |
# device_map="balanced",
|