From ca59e80a75b57b143994d0b21b7a46db4f65a924 Mon Sep 17 00:00:00 2001
From: William Falcon <waf2107@columbia.edu>
Date: Sun, 14 Apr 2024 16:09:04 -0400
Subject: [PATCH] Update README.md

Adds "Open in Studio" badge.

Looks like someone implemented this into a Lightning Studio so it can be opened and run on the cloud!
---
 README.md | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index fea53e6..09b1058 100644
--- a/README.md
+++ b/README.md
@@ -1,8 +1,13 @@
 # BitNet-1.58-Instruct
 
+<p align="center" style="display: flex; direction:row; align-items: center">
 <a href="https://docs.oxen.ai/" style="padding: 2px;">
     <img src="https://img.shields.io/endpoint?url=https://hub.oxen.ai/api/repos/ox/BitNet/badge">
+</a>   
+<a target="_blank" href="https://lightning.ai/someshfengde/studios/1-5bitllms-finetuning">
+  <img src="https://pl-bolts-doc-images.s3.us-east-2.amazonaws.com/app-2/studio-badge.svg" alt="Open In Studio"/>
 </a>
+</p>
 
 Implementation of BitNet-1.58 instruct tuning. All data and models are versioned and stored on [Oxen.ai](https://Oxen.ai/ox/BitNet) at [ox/BitNet](https://Oxen.ai/ox/BitNet). This work builds off the pre-trained models released in the [1bitLLM/bitnet_b1_58-large](https://huggingface.co/1bitLLM/bitnet_b1_58-large) project on Hugging Face.
 
@@ -99,4 +104,4 @@ The eval script outputs a dataframe like this:
 
 ```
 TODO:
-```
\ No newline at end of file
+```