bugfix
Browse files- app.py +1 -1
- requirements.txt +2 -1
app.py
CHANGED
@@ -23,7 +23,7 @@ login(token=HF_TOKEN)
|
|
23 |
|
24 |
bfl_repo = "black-forest-labs/FLUX.1-dev"
|
25 |
|
26 |
-
pipe = FluxPipeline.from_pretrained("jiuface/lover-snap
|
27 |
# pipe.transformer = transformer
|
28 |
# pipe.text_encoder_2 = text_encoder_2
|
29 |
|
|
|
23 |
|
24 |
bfl_repo = "black-forest-labs/FLUX.1-dev"
|
25 |
|
26 |
+
pipe = FluxPipeline.from_pretrained("jiuface/lover-snap", use_safetensors=True, torch_dtype=torch.bfloat16)
|
27 |
# pipe.transformer = transformer
|
28 |
# pipe.text_encoder_2 = text_encoder_2
|
29 |
|
requirements.txt
CHANGED
@@ -4,4 +4,5 @@ spaces
|
|
4 |
transformers
|
5 |
peft
|
6 |
sentencepiece
|
7 |
-
optimum-quanto
|
|
|
|
4 |
transformers
|
5 |
peft
|
6 |
sentencepiece
|
7 |
+
optimum-quanto
|
8 |
+
safetensors
|