Hugging Face's logo

Spaces:
YuLexuan30
/
Chroma_latest
Paused

App Files Community
gokaygokay commited on
Commit
c83ee24
·
2 Parent(s): 7cda016 d56808a

Merge branch 'main' of https://huggingface.co/spaces/gokaygokay/Video-Prompt-Generator

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +1 -0
app.py CHANGED
@@ -7,7 +7,7 @@ from transformers import T5EncoderModel, BitsAndBytesConfig as BitsAndBytesConfi
7
  # Initialize model outside the function
8
  device = "cuda" if torch.cuda.is_available() else "cpu"
9
  dtype = torch.bfloat16
10
- base_model = "lodestones/Chroma"
11
  file_url = "https://huggingface.co/lodestones/Chroma/resolve/main/chroma-unlocked-v31.safetensors"
12
 
13
  quantization_config_tf = BitsAndBytesConfigTF(load_in_8bit=True, bnb_8bit_compute_dtype=torch.bfloat16)
 
7
  # Initialize model outside the function
8
  device = "cuda" if torch.cuda.is_available() else "cpu"
9
  dtype = torch.bfloat16
10
+ single_file_base_model = "camenduru/FLUX.1-dev-diffusers"
11
  file_url = "https://huggingface.co/lodestones/Chroma/resolve/main/chroma-unlocked-v31.safetensors"
12
 
13
  quantization_config_tf = BitsAndBytesConfigTF(load_in_8bit=True, bnb_8bit_compute_dtype=torch.bfloat16)
requirements.txt CHANGED
@@ -5,4 +5,5 @@ gradio
5
  transformers
6
  xformers
7
  sentencepiece
 
8
  peft
 
5
  transformers
6
  xformers
7
  sentencepiece
8
+ bitsandbytes
9
  peft