Upload 2 files

app.py

@@ -14,15 +14,17 @@ from typing import List, Union
 import torchvision.transforms.functional as TVF
 from peft import PeftModel
 import gc
+import sys
+IS_COLAB = 'google.colab' in sys.modules
 
 # Constants
 IMAGE_EXTENSIONS = ('.jpg', '.jpeg', '.png', '.bmp', '.webp')
 HF_TOKEN = os.environ.get("HF_TOKEN", None)
 BASE_DIR = Path(__file__).resolve().parent # Define the base directory
+CHECKPOINT_PATH = BASE_DIR / Path("cgrkzexw-599808") if not IS_COLAB else Path("/content/joy-caption-alpha-two-cli-mod/cgrkzexw-599808")
 CLIP_PATH = "google/siglip-so400m-patch14-384"
 DEFAULT_MODEL_PATH = "unsloth/Meta-Llama-3.1-8B-bnb-4bit"
 #DEFAULT_MODEL_PATH = "Orenguteng/Llama-3.1-8B-Lexi-Uncensored-V2" # Works better but full weight.
-CHECKPOINT_PATH = BASE_DIR / Path("cgrkzexw-599808")
 LORA_PATH = CHECKPOINT_PATH / "text_model"
 CAPTION_TYPE_MAP = {
     "Descriptive": [

joycaption_alpha_two_cli_mod.ipynb

@@ -0,0 +1,45 @@
+{
+  "cells": [
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "ZgkQ4kDil23W"
+      },
+      "outputs": [],
+      "source": [
+        "!git clone https://huggingface.co/John6666/joy-caption-alpha-two-cli-mod/\n",
+        "!pip install -r /content/joy-caption-alpha-two-cli-mod/requirements.txt\n",
+        "!pip install bitsandbytes triton\n",
+        "!python /content/joy-caption-alpha-two-cli-mod/app.py"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "gPwD8BVsnU7p"
+      },
+      "outputs": [],
+      "source": [
+        "!python /content/joy-caption-alpha-two-cli-mod/app.py"
+      ]
+    }
+  ],
+  "metadata": {
+    "accelerator": "GPU",
+    "colab": {
+      "gpuType": "T4",
+      "provenance": []
+    },
+    "kernelspec": {
+      "display_name": "Python 3",
+      "name": "python3"
+    },
+    "language_info": {
+      "name": "python"
+    }
+  },
+  "nbformat": 4,
+  "nbformat_minor": 0
+}