first

.idea/.gitignore

@@ -0,0 +1,3 @@
+# Default ignored files
+/shelf/
+/workspace.xml

.idea/Gemma_2B_DataScience_RAFT_QA.iml

@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<module type="PYTHON_MODULE" version="4">
+  <component name="NewModuleRootManager">
+    <content url="file://$MODULE_DIR$" />
+    <orderEntry type="inheritedJdk" />
+    <orderEntry type="sourceFolder" forTests="false" />
+  </component>
+</module>

.idea/inspectionProfiles/Project_Default.xml

@@ -0,0 +1,207 @@
+<component name="InspectionProjectProfileManager">
+  <profile version="1.0">
+    <option name="myName" value="Project Default" />
+    <inspection_tool class="PyPackageRequirementsInspection" enabled="true" level="WARNING" enabled_by_default="true">
+      <option name="ignoredPackages">
+        <value>
+          <list size="187">
+            <item index="0" class="java.lang.String" itemvalue="importlib-metadata" />
+            <item index="1" class="java.lang.String" itemvalue="chromadb" />
+            <item index="2" class="java.lang.String" itemvalue="tzlocal" />
+            <item index="3" class="java.lang.String" itemvalue="scikit-learn" />
+            <item index="4" class="java.lang.String" itemvalue="tabulate" />
+            <item index="5" class="java.lang.String" itemvalue="opentelemetry-proto" />
+            <item index="6" class="java.lang.String" itemvalue="validators" />
+            <item index="7" class="java.lang.String" itemvalue="ufw" />
+            <item index="8" class="java.lang.String" itemvalue="pyth" />
+            <item index="9" class="java.lang.String" itemvalue="pycairo" />
+            <item index="10" class="java.lang.String" itemvalue="pycparser" />
+            <item index="11" class="java.lang.String" itemvalue="cmdtest" />
+            <item index="12" class="java.lang.String" itemvalue="PyGObject" />
+            <item index="13" class="java.lang.String" itemvalue="posthog" />
+            <item index="14" class="java.lang.String" itemvalue="distro-info" />
+            <item index="15" class="java.lang.String" itemvalue="pyasn1-modules" />
+            <item index="16" class="java.lang.String" itemvalue="systemd-python" />
+            <item index="17" class="java.lang.String" itemvalue="cliapp" />
+            <item index="18" class="java.lang.String" itemvalue="language-selector" />
+            <item index="19" class="java.lang.String" itemvalue="uvicorn" />
+            <item index="20" class="java.lang.String" itemvalue="pysimdjson" />
+            <item index="21" class="java.lang.String" itemvalue="terminado" />
+            <item index="22" class="java.lang.String" itemvalue="pydantic" />
+            <item index="23" class="java.lang.String" itemvalue="Werkzeug" />
+            <item index="24" class="java.lang.String" itemvalue="pyglet" />
+            <item index="25" class="java.lang.String" itemvalue="streamlit" />
+            <item index="26" class="java.lang.String" itemvalue="opentelemetry-api" />
+            <item index="27" class="java.lang.String" itemvalue="pinecone-client" />
+            <item index="28" class="java.lang.String" itemvalue="reportlab" />
+            <item index="29" class="java.lang.String" itemvalue="PyMuPDFb" />
+            <item index="30" class="java.lang.String" itemvalue="pexpect" />
+            <item index="31" class="java.lang.String" itemvalue="pypdf" />
+            <item index="32" class="java.lang.String" itemvalue="python-apt" />
+            <item index="33" class="java.lang.String" itemvalue="SecretStorage" />
+            <item index="34" class="java.lang.String" itemvalue="ubuntu-advantage-tools" />
+            <item index="35" class="java.lang.String" itemvalue="opentelemetry-util-http" />
+            <item index="36" class="java.lang.String" itemvalue="psutil" />
+            <item index="37" class="java.lang.String" itemvalue="openai" />
+            <item index="38" class="java.lang.String" itemvalue="pyxdg" />
+            <item index="39" class="java.lang.String" itemvalue="nvidia-cublas-cu12" />
+            <item index="40" class="java.lang.String" itemvalue="paramiko" />
+            <item index="41" class="java.lang.String" itemvalue="pydantic_core" />
+            <item index="42" class="java.lang.String" itemvalue="platformdirs" />
+            <item index="43" class="java.lang.String" itemvalue="referencing" />
+            <item index="44" class="java.lang.String" itemvalue="PyJWT" />
+            <item index="45" class="java.lang.String" itemvalue="netifaces" />
+            <item index="46" class="java.lang.String" itemvalue="nvidia-nvjitlink-cu12" />
+            <item index="47" class="java.lang.String" itemvalue="nvidia-cusparse-cu12" />
+            <item index="48" class="java.lang.String" itemvalue="pandocfilters" />
+            <item index="49" class="java.lang.String" itemvalue="py4j" />
+            <item index="50" class="java.lang.String" itemvalue="pyasn1" />
+            <item index="51" class="java.lang.String" itemvalue="jupyter-events" />
+            <item index="52" class="java.lang.String" itemvalue="nvidia-nccl-cu12" />
+            <item index="53" class="java.lang.String" itemvalue="sniffio" />
+            <item index="54" class="java.lang.String" itemvalue="websocket-client" />
+            <item index="55" class="java.lang.String" itemvalue="seaborn" />
+            <item index="56" class="java.lang.String" itemvalue="zipp" />
+            <item index="57" class="java.lang.String" itemvalue="prompt-toolkit" />
+            <item index="58" class="java.lang.String" itemvalue="xgboost" />
+            <item index="59" class="java.lang.String" itemvalue="together" />
+            <item index="60" class="java.lang.String" itemvalue="websockets" />
+            <item index="61" class="java.lang.String" itemvalue="pyarrow" />
+            <item index="62" class="java.lang.String" itemvalue="PyOpenGL" />
+            <item index="63" class="java.lang.String" itemvalue="ubuntu-drivers-common" />
+            <item index="64" class="java.lang.String" itemvalue="PyMuPDF" />
+            <item index="65" class="java.lang.String" itemvalue="scipy" />
+            <item index="66" class="java.lang.String" itemvalue="watchfiles" />
+            <item index="67" class="java.lang.String" itemvalue="tornado" />
+            <item index="68" class="java.lang.String" itemvalue="xkit" />
+            <item index="69" class="java.lang.String" itemvalue="nvidia-cudnn-cu12" />
+            <item index="70" class="java.lang.String" itemvalue="torch" />
+            <item index="71" class="java.lang.String" itemvalue="overrides" />
+            <item index="72" class="java.lang.String" itemvalue="toml" />
+            <item index="73" class="java.lang.String" itemvalue="opentelemetry-semantic-conventions" />
+            <item index="74" class="java.lang.String" itemvalue="pandas" />
+            <item index="75" class="java.lang.String" itemvalue="watchdog" />
+            <item index="76" class="java.lang.String" itemvalue="typing_extensions" />
+            <item index="77" class="java.lang.String" itemvalue="dbus-python" />
+            <item index="78" class="java.lang.String" itemvalue="olefile" />
+            <item index="79" class="java.lang.String" itemvalue="usb-creator" />
+            <item index="80" class="java.lang.String" itemvalue="opentelemetry-instrumentation" />
+            <item index="81" class="java.lang.String" itemvalue="traitlets" />
+            <item index="82" class="java.lang.String" itemvalue="protobuf" />
+            <item index="83" class="java.lang.String" itemvalue="ttystatus" />
+            <item index="84" class="java.lang.String" itemvalue="pygame" />
+            <item index="85" class="java.lang.String" itemvalue="threadpoolctl" />
+            <item index="86" class="java.lang.String" itemvalue="Brlapi" />
+            <item index="87" class="java.lang.String" itemvalue="duplicity" />
+            <item index="88" class="java.lang.String" itemvalue="python-dateutil" />
+            <item index="89" class="java.lang.String" itemvalue="roman" />
+            <item index="90" class="java.lang.String" itemvalue="pycups" />
+            <item index="91" class="java.lang.String" itemvalue="python-dotenv" />
+            <item index="92" class="java.lang.String" itemvalue="gym-maze" />
+            <item index="93" class="java.lang.String" itemvalue="PyPika" />
+            <item index="94" class="java.lang.String" itemvalue="nvidia-cusolver-cu12" />
+            <item index="95" class="java.lang.String" itemvalue="pyzmq" />
+            <item index="96" class="java.lang.String" itemvalue="polars" />
+            <item index="97" class="java.lang.String" itemvalue="pyparsing" />
+            <item index="98" class="java.lang.String" itemvalue="unattended-upgrades" />
+            <item index="99" class="java.lang.String" itemvalue="opentelemetry-instrumentation-asgi" />
+            <item index="100" class="java.lang.String" itemvalue="sympy" />
+            <item index="101" class="java.lang.String" itemvalue="uvloop" />
+            <item index="102" class="java.lang.String" itemvalue="triton" />
+            <item index="103" class="java.lang.String" itemvalue="setuptools-git" />
+            <item index="104" class="java.lang.String" itemvalue="opentelemetry-exporter-otlp-proto-grpc" />
+            <item index="105" class="java.lang.String" itemvalue="pulsar-client" />
+            <item index="106" class="java.lang.String" itemvalue="orjson" />
+            <item index="107" class="java.lang.String" itemvalue="opentelemetry-instrumentation-fastapi" />
+            <item index="108" class="java.lang.String" itemvalue="duckdb" />
+            <item index="109" class="java.lang.String" itemvalue="nvidia-nvtx-cu12" />
+            <item index="110" class="java.lang.String" itemvalue="streamlit-float" />
+            <item index="111" class="java.lang.String" itemvalue="widgetsnbextension" />
+            <item index="112" class="java.lang.String" itemvalue="gyp" />
+            <item index="113" class="java.lang.String" itemvalue="pyautogen" />
+            <item index="114" class="java.lang.String" itemvalue="webargs" />
+            <item index="115" class="java.lang.String" itemvalue="whatsapp-python" />
+            <item index="116" class="java.lang.String" itemvalue="miniworld" />
+            <item index="117" class="java.lang.String" itemvalue="defer" />
+            <item index="118" class="java.lang.String" itemvalue="pydeck" />
+            <item index="119" class="java.lang.String" itemvalue="ptyprocess" />
+            <item index="120" class="java.lang.String" itemvalue="SQLAlchemy" />
+            <item index="121" class="java.lang.String" itemvalue="opentelemetry-sdk" />
+            <item index="122" class="java.lang.String" itemvalue="pyspark" />
+            <item index="123" class="java.lang.String" itemvalue="wcwidth" />
+            <item index="124" class="java.lang.String" itemvalue="types-python-dateutil" />
+            <item index="125" class="java.lang.String" itemvalue="python-debian" />
+            <item index="126" class="java.lang.String" itemvalue="cupshelpers" />
+            <item index="127" class="java.lang.String" itemvalue="onnxruntime" />
+            <item index="128" class="java.lang.String" itemvalue="chroma-hnswlib" />
+            <item index="129" class="java.lang.String" itemvalue="rpds-py" />
+            <item index="130" class="java.lang.String" itemvalue="tomli" />
+            <item index="131" class="java.lang.String" itemvalue="louis" />
+            <item index="132" class="java.lang.String" itemvalue="urllib3" />
+            <item index="133" class="java.lang.String" itemvalue="faiss-gpu" />
+            <item index="134" class="java.lang.String" itemvalue="sseclient-py" />
+            <item index="135" class="java.lang.String" itemvalue="typer" />
+            <item index="136" class="java.lang.String" itemvalue="pandasai" />
+            <item index="137" class="java.lang.String" itemvalue="parso" />
+            <item index="138" class="java.lang.String" itemvalue="prometheus-client" />
+            <item index="139" class="java.lang.String" itemvalue="pyth3" />
+            <item index="140" class="java.lang.String" itemvalue="pillow" />
+            <item index="141" class="java.lang.String" itemvalue="xdg" />
+            <item index="142" class="java.lang.String" itemvalue="command-not-found" />
+            <item index="143" class="java.lang.String" itemvalue="opentelemetry-exporter-otlp-proto-common" />
+            <item index="144" class="java.lang.String" itemvalue="accelerate" />
+            <item index="145" class="java.lang.String" itemvalue="diffusers" />
+            <item index="146" class="java.lang.String" itemvalue="gradio" />
+            <item index="147" class="java.lang.String" itemvalue="huggingface-hub" />
+            <item index="148" class="java.lang.String" itemvalue="PyYAML" />
+            <item index="149" class="java.lang.String" itemvalue="MarkupSafe" />
+            <item index="150" class="java.lang.String" itemvalue="requests" />
+            <item index="151" class="java.lang.String" itemvalue="Jinja2" />
+            <item index="152" class="java.lang.String" itemvalue="markdown-it-py" />
+            <item index="153" class="java.lang.String" itemvalue="fsspec" />
+            <item index="154" class="java.lang.String" itemvalue="jsonschema-specifications" />
+            <item index="155" class="java.lang.String" itemvalue="semantic-version" />
+            <item index="156" class="java.lang.String" itemvalue="tomlkit" />
+            <item index="157" class="java.lang.String" itemvalue="filelock" />
+            <item index="158" class="java.lang.String" itemvalue="gradio_client" />
+            <item index="159" class="java.lang.String" itemvalue="Pygments" />
+            <item index="160" class="java.lang.String" itemvalue="safetensors" />
+            <item index="161" class="java.lang.String" itemvalue="starlette" />
+            <item index="162" class="java.lang.String" itemvalue="mdurl" />
+            <item index="163" class="java.lang.String" itemvalue="jsonschema" />
+            <item index="164" class="java.lang.String" itemvalue="tokenizers" />
+            <item index="165" class="java.lang.String" itemvalue="transformers" />
+            <item index="166" class="java.lang.String" itemvalue="importlib_resources" />
+            <item index="167" class="java.lang.String" itemvalue="aiofiles" />
+            <item index="168" class="java.lang.String" itemvalue="rich" />
+            <item index="169" class="java.lang.String" itemvalue="pydub" />
+            <item index="170" class="java.lang.String" itemvalue="python-multipart" />
+            <item index="171" class="java.lang.String" itemvalue="click" />
+            <item index="172" class="java.lang.String" itemvalue="altair" />
+            <item index="173" class="java.lang.String" itemvalue="attrs" />
+            <item index="174" class="java.lang.String" itemvalue="fastapi" />
+            <item index="175" class="java.lang.String" itemvalue="regex" />
+            <item index="176" class="java.lang.String" itemvalue="toolz" />
+            <item index="177" class="java.lang.String" itemvalue="ruff" />
+            <item index="178" class="java.lang.String" itemvalue="charset-normalizer" />
+            <item index="179" class="java.lang.String" itemvalue="ffmpy" />
+            <item index="180" class="java.lang.String" itemvalue="shellingham" />
+            <item index="181" class="java.lang.String" itemvalue="langchain" />
+            <item index="182" class="java.lang.String" itemvalue="sentence-transformers" />
+            <item index="183" class="java.lang.String" itemvalue="Pillow" />
+            <item index="184" class="java.lang.String" itemvalue="langchain-text-splitters" />
+            <item index="185" class="java.lang.String" itemvalue="langchain_core" />
+            <item index="186" class="java.lang.String" itemvalue="langchain_community" />
+          </list>
+        </value>
+      </option>
+    </inspection_tool>
+    <inspection_tool class="PyUnresolvedReferencesInspection" enabled="true" level="WARNING" enabled_by_default="true">
+      <option name="ignoredIdentifiers">
+        <list>
+          <option value="spaces" />
+        </list>
+      </option>
+    </inspection_tool>
+  </profile>
+</component>

.idea/inspectionProfiles/profiles_settings.xml

@@ -0,0 +1,6 @@
+<component name="InspectionProjectProfileManager">
+  <settings>
+    <option name="USE_PROJECT_PROFILE" value="false" />
+    <version value="1.0" />
+  </settings>
+</component>

.idea/modules.xml

@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectModuleManager">
+    <modules>
+      <module fileurl="file://$PROJECT_DIR$/.idea/Gemma_2B_DataScience_RAFT_QA.iml" filepath="$PROJECT_DIR$/.idea/Gemma_2B_DataScience_RAFT_QA.iml" />
+    </modules>
+  </component>
+</project>

.idea/vcs.xml

@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="VcsDirectoryMappings">
+    <mapping directory="" vcs="Git" />
+  </component>
+</project>

app.py

@@ -0,0 +1,55 @@
+import gradio as gr
+import spaces
+import torch
+
+import transformers
+import torch
+from transformers import AutoModelForCausalLM, AutoTokenizer
+
+model_name = "ModularityAI/gemma-2b-datascience-it-raft"
+
+pipeline = transformers.pipeline(
+    "text-generation",
+    model=model_name,
+    model_kwargs={"torch_dtype": torch.bfloat16},
+    device="cuda",
+)
+
+@spaces.GPU
+def chat_function(message, history, system_prompt,max_new_tokens,temperature):
+    messages = [
+        {"role": "system", "content": system_prompt},
+        {"role": "user", "content": message},
+    ]
+    prompt = pipeline.tokenizer.apply_chat_template(
+        messages,
+        tokenize=False,
+        add_generation_prompt=True
+    )
+    print(prompt)
+    temp = temperature + 0.1
+    outputs = pipeline(
+        prompt,
+        max_new_tokens=max_new_tokens,
+        do_sample=True,
+        temperature=temp,
+        top_p=0.9,
+    )
+    return outputs[0]["generated_text"][len(prompt):]
+
+gr.ChatInterface(
+    chat_function,
+    chatbot=gr.Chatbot(height=400),
+    textbox=gr.Textbox(placeholder="Enter message here", container=False, scale=7),
+    title="LLAMA 3 8B Chat",
+    description="""
+    This space is dedicated for chatting with Gemma 2B Finetuned for Data Science QA using RAFT. Find this model here: https://huggingface.co/ModularityAI/gemma-2b-datascience-it-raft
+    Feel free to play with customization in the "Additional Inputs".
+    """,
+    theme="soft",
+    additional_inputs=[
+        gr.Textbox("You are helpful AI.", label="System Prompt"),
+        gr.Slider(512, 4096, label="Max New Tokens"),
+        gr.Slider(0, 1, label="Temperature")
+    ]
+).launch()

requirements.txt

@@ -0,0 +1,78 @@
+aiofiles==23.2.1
+altair==5.3.0
+annotated-types==0.6.0
+anyio==4.3.0
+attrs==23.2.0
+certifi==2024.2.2
+charset-normalizer==3.3.2
+click==8.1.7
+colorama==0.4.6
+contourpy==1.2.1
+cycler==0.12.1
+distro==1.9.0
+fastapi==0.110.1
+ffmpy==0.3.2
+filelock==3.13.4
+fonttools==4.51.0
+fsspec==2024.3.1
+gradio==4.26.0
+gradio_client==0.15.1
+h11==0.14.0
+httpcore==1.0.5
+httpx==0.27.0
+huggingface-hub==0.22.2
+idna==3.6
+importlib_resources==6.4.0
+Jinja2==3.1.3
+joblib==1.3.2
+jsonschema==4.21.1
+jsonschema-specifications==2023.12.1
+kiwisolver==1.4.5
+markdown-it-py==3.0.0
+MarkupSafe==2.1.5
+matplotlib==3.8.4
+mdurl==0.1.2
+numpy==1.26.4
+openai==1.16.2
+orjson==3.10.0
+packaging==24.0
+pandas==2.2.1
+pillow==10.3.0
+pydantic==2.6.4
+pydantic_core==2.16.3
+pydub==0.25.1
+Pygments==2.17.2
+pyparsing==3.1.2
+python-dateutil==2.9.0.post0
+python-multipart==0.0.9
+pytz==2024.1
+PyYAML==6.0.1
+referencing==0.34.0
+regex==2023.12.25
+requests==2.31.0
+rich==13.7.1
+rpds-py==0.18.0
+ruff==0.3.5
+safetensors==0.4.2
+scikit-learn==1.4.1.post1
+scipy==1.13.0
+semantic-version==2.10.0
+shellingham==1.5.4
+six==1.16.0
+sniffio==1.3.1
+starlette==0.37.2
+threadpoolctl==3.4.0
+tokenizers==0.15.2
+tomlkit==0.12.0
+toolz==0.12.1
+tqdm==4.66.2
+transformers==4.39.3
+typer==0.12.3
+typing_extensions==4.11.0
+tzdata==2024.1
+urllib3==2.2.1
+uvicorn==0.29.0
+websockets==11.0.3
+xgboost==2.0.3
+accelerate==0.29.2
+diffusers==0.27.2