Spaces:
Runtime error
Runtime error
change flux model to lite one
Browse files
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 |
+
"Freepik/flux.1-lite-8B-alpha",
|
| 59 |
torch_dtype=torch.bfloat16, # Используем bfloat16 для A100
|
| 60 |
# low_cpu_mem_usage=True, # Экономия памяти
|
| 61 |
# device_map="balanced",
|