Update README.md

README.md

@@ -45,8 +45,11 @@ Augmentations used : Center crop, Random Flip
 Gradient accumulation steps : Tried 1, 3, and 4 for different experiments. 4 gave decent results.
 
 text2image_fine-tune :
+
 **wandb dashboard : https://wandb.ai/asrimanth/text2image_fine-tune**
+
 **Model card for asrimanth/person-thumbs-up-lora: https://huggingface.co/asrimanth/person-thumbs-up-lora**
+
 **Prompt: ```<tom_cruise> #thumbsup```**
 
 Deployed models:
@@ -54,18 +57,27 @@ Deployed models:
 When the above experiment failed, I had to try different datasets. One of them was "tom cruise".
 
 srimanth-thumbs-up-lora-plain : We use the plain dataset with srimanth mentioned above.
+
 **wandb link: https://wandb.ai/asrimanth/srimanth-thumbs-up-lora-plain**
+
 **Model card for srimanth-thumbs-up-lora-plain: https://huggingface.co/asrimanth/srimanth-thumbs-up-lora-plain**
+
 **Prompt: ```srimanth thumbs up```**
 
-person-thumbs-up-plain-lora wandb : We use the 
+person-thumbs-up-plain-lora wandb : We use the plain dataset with tom cruise images.
+
 **wandb link: https://wandb.ai/asrimanth/person-thumbs-up-plain-lora**
+
 **Model card for asrimanth/person-thumbs-up-plain-lora: https://huggingface.co/asrimanth/person-thumbs-up-plain-lora**
+
 **Prompt: ```tom cruise thumbs up```**
 
-person-thumbs-up-lora-no-cap wandb dashboard:
+person-thumbs-up-lora-no-cap wandb dashboard: We use the no-cap dataset with tom cruise images.
+
 **https://wandb.ai/asrimanth/person-thumbs-up-lora-no-cap**
+
 **Model card for asrimanth/person-thumbs-up-lora-no-cap: https://huggingface.co/asrimanth/person-thumbs-up-lora-no-cap**
+
 **Prompt: ```<tom_cruise> #thumbsup```**
 
 ### Inference
@@ -75,6 +87,7 @@ person-thumbs-up-lora-no-cap wandb dashboard:
 + During local inference (25 steps), I found the person-thumbs-up-plain-lora to show 35 out of 50 images with a decent thumbs up result for tom cruise, 5 incomplete thumbs up.
 + While I could not evaluate the model with metrics due to insufficient time, I chose the visual approach. To view the inference images, check the `results` folder.
 + To evaulate diffusion models, I would start with this: https://huggingface.co/docs/diffusers/conceptual/evaluation
++ The half-precision inference was not working on CPU, so we've used torch.float32 instead.
 
 ### Deployment