chroma

app.py

@@ -7,7 +7,7 @@ from transformers import T5EncoderModel, BitsAndBytesConfig as BitsAndBytesConfi
 # Initialize model outside the function
 device = "cuda" if torch.cuda.is_available() else "cpu"
 dtype = torch.bfloat16
-single_file_base_model = "camenduru/FLUX.1-dev-diffusers"
+base_model = "lodestones/Chroma"
 file_url = "https://huggingface.co/lodestones/Chroma/resolve/main/chroma-unlocked-v31.safetensors"
 
 quantization_config_tf = BitsAndBytesConfigTF(load_in_8bit=True, bnb_8bit_compute_dtype=torch.bfloat16)
@@ -42,7 +42,7 @@ iface = gr.Interface(
     ],
     outputs=gr.Image(label="Generated Image"),
     title="Chroma Image Generator",
-    description="Generate images using the Chroma model with FLUX pipeline",
+    description="Generate images using the Chroma model",
     examples=[
         ["A beautiful sunset over mountains, photorealistic, 8k", "blurry, low quality, distorted", 30, 7.5],
         ["A futuristic cityscape at night, neon lights, cyberpunk style", "ugly, deformed, low resolution", 30, 7.5]