Update README.md

README.md

@@ -81,6 +81,11 @@ output = _postprocess_output_cypher(raw_output)
 print(output)
 ```
 
+### NOTE on creating your own schemas:
+# In the dataset we used, the schemas are already provided.
+# They are created either by Directly using the schema the input data source provided OR
+# Creating schema using neo4j-graphrag package (Check: SchemaReader.get_schema(...) function)
+# In your own Neo4j database, you can utilize neo4j-graphrag package::SchemaReader functions
 
 
 ## Bias, Risks, and Limitations
@@ -105,8 +110,3 @@ lora_config = LoraConfig( r=64, lora_alpha=64, target_modules=target_modules, lo
 sft_config = SFTConfig( dataset_text_field=dataset_text_field, per_device_train_batch_size=4, gradient_accumulation_steps=8, dataset_num_proc=16, max_seq_length=1600, logging_dir="./logs", num_train_epochs=1, learning_rate=2e-5, save_steps=5, save_total_limit=1, logging_steps=5, output_dir="outputs", optim="paged_adamw_8bit", save_strategy="steps", )
 bnb_config = BitsAndBytesConfig( load_in_4bit=True, bnb_4bit_use_double_quant=True, bnb_4bit_quant_type="nf4", bnb_4bit_compute_dtype=torch.bfloat16, )
 
-### NOTE on creating your own schemas:
-In the dataset we used, the schemas are already provided. They are created either by
-Directly using the schema the input data source provided OR
-Creating schema using neo4j-graphrag package (Check: SchemaReader.get_schema(...) function)
-In your own Neo4j database, you can utilize neo4j-graphrag package::SchemaReader functions