update readme and reformat

Author: leondgarse

README.md

@@ -1,7 +1,8 @@
 # ___Keras_cv_attention_models___
 ***
-- **WARNING: currently NOT compatible with `keras 3.0`, if using `tensorflow>=2.16.0`, needs to install `pip install tf-keras~=2.16` manually.**
-- **coco_train_script.py is under testing. Still struggling for this...**
+- **WARNING: currently NOT compatible with `keras 3.0`, if using `tensorflow>=2.16.0`, needs to install `pip install tf-keras~=2.16` manually. While importing, import this package ahead of Tensorflow, or set `export TF_USE_LEGACY_KERAS=1`.**
+- **It's not recommended downloading and loading model from h5 file directly, better building model and loading weights like `import kecam; mm = kecam.models.LCNet050()`.**
+- **coco_train_script.py for TF is still under testing...**
 <!-- TOC depthFrom:1 depthTo:6 withLinks:1 updateOnSave:1 orderedList:0 -->
 
 - [___>>>> Roadmap and todo list <<<<___](https://github.com/leondgarse/keras_cv_attention_models/wiki/Roadmap)
@@ -109,7 +110,6 @@
 
 # General Usage
 ## Basic
-  - **Currently recommended TF version is `tensorflow==2.11.1`. Expecially for training or TFLite conversion**.
   - **Default import** will not specific these while using them in READMEs.
     ```py
     import os

keras_cv_attention_models/llama2/llama2.py

@@ -158,7 +158,7 @@ def causal_self_attention_with_cache(inputs, start_pos=0, max_batch_size=0, bloc
     key = layers.Reshape([-1, num_kv_heads, key_dim], name=name + "key_reshape")(key)
     key = apply_positional_encoding_rotary(key, rope, start_pos, is_kv_cache, name=name + "key_")
     if is_kv_cache:  # Use KV cache if max_batch_size specified
-        key = KVCache(max_batch_size=max_batch_size, max_seq_len=block_size, name=name + "key_cahce")([key, start_pos])
+        key = KVCache(max_batch_size=max_batch_size, max_seq_len=block_size, name=name + "key_cache")([key, start_pos])
     key = functional.transpose(key, [0, 2, 3, 1])
 
     value = layers.Reshape([-1, num_kv_heads, key_dim], name=name + "value_reshape")(value)

setup.py

@@ -1,5 +1,6 @@
 """ Setup
 """
+
 from setuptools import setup, find_packages
 from codecs import open
 from os import path

setup_kecam.py

@@ -1,5 +1,6 @@
 """ Setup
 """
+
 from setuptools import setup, find_packages
 from codecs import open
 from os import path

tests/test_layers.py

@@ -1,4 +1,5 @@
 import sys
+
 sys.path.append(".")
 import keras_cv_attention_models  # Needs to set TF_USE_LEGACY_KERAS=1 env firstly
 

tests/test_models.py

@@ -1,4 +1,5 @@
 import sys
+
 sys.path.append(".")
 import keras_cv_attention_models  # Needs to set TF_USE_LEGACY_KERAS=1 env firstly
 

tests/test_models_tf.py

@@ -1,4 +1,5 @@
 import sys
+
 sys.path.append(".")
 import keras_cv_attention_models  # Needs to set TF_USE_LEGACY_KERAS=1 env firstly
 

tests/test_switch_to_deploy_tf.py

@@ -1,4 +1,5 @@
 import sys
+
 sys.path.append(".")
 import keras_cv_attention_models  # Needs to set TF_USE_LEGACY_KERAS=1 env firstly