From a04d139ff5b56ba2eb26e3e481a26b69431a7ee5 Mon Sep 17 00:00:00 2001
From: Fabio <f.xuzzo@gmail.com>
Date: Sat, 14 Sep 2024 20:04:08 +0200
Subject: [PATCH] further cleanup

---
 src/mflux/config/config.py         |  2 +-
 src/mflux/config/runtime_config.py |  2 --
 trial.py                           | 38 ------------------------------
 3 files changed, 1 insertion(+), 41 deletions(-)
 delete mode 100644 trial.py

diff --git a/src/mflux/config/config.py b/src/mflux/config/config.py
index 88566ae..123bb44 100644
--- a/src/mflux/config/config.py
+++ b/src/mflux/config/config.py
@@ -6,7 +6,7 @@ log = logging.getLogger(__name__)
 
 
 class Config:
-    precision: mx.Dtype = mx.float16
+    precision: mx.Dtype = mx.bfloat16
 
     def __init__(
             self,
diff --git a/src/mflux/config/runtime_config.py b/src/mflux/config/runtime_config.py
index dabd8f5..96bd89b 100644
--- a/src/mflux/config/runtime_config.py
+++ b/src/mflux/config/runtime_config.py
@@ -1,5 +1,3 @@
-from math import e
-from cycler import V
 import mlx.core as mx
 import numpy as np
 
diff --git a/trial.py b/trial.py
deleted file mode 100644
index ee933b2..0000000
--- a/trial.py
+++ /dev/null
@@ -1,38 +0,0 @@
-import os
-import sys
-from mflux.post_processing.image_util import ImageUtil
-
-sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
-
-from mflux.config.model_config import ModelConfig
-from mflux.config.config import ConfigControlnet
-from mflux.controlnet.flux_controlnet import Flux1Controlnet
-
-prompt = "Luxury picture of food"
-
-
-# Load the model
-flux = Flux1Controlnet(
-    model_config=ModelConfig.from_alias("dev"),
-    quantize=8,
-    lora_paths=["Flux_1_Dev_LoRA_Paper-Cutout-Style.safetensors"]
-)
-
-control_image = ImageUtil.load_image("image_1.png")
-
-# Generate an image
-image = flux.generate_image(
-    seed=2500,
-    prompt=prompt,
-    control_image=control_image,
-    config=ConfigControlnet(
-        num_inference_steps=10,
-        height=256,
-        width=512,
-        guidance=3.5,
-        controlnet_strength=0.7,
-    )
-)
-
-# Save the image
-image.save(path="image.png", export_json_metadata=False)
\ No newline at end of file