Update app.py
Browse files
app.py
CHANGED
|
@@ -46,7 +46,7 @@ repo_nsfw_classifier = pipeline("image-classification", model="Falconsai/nsfw_im
|
|
| 46 |
# repo_default.load_lora_weights("artificialguybr/PixelArtRedmond", adapter_name="pixel_base")
|
| 47 |
# repo_default.load_lora_weights("nerijs/pixel-art-xl", adapter_name="pixel_base_2")
|
| 48 |
|
| 49 |
-
repo_large = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
|
| 50 |
# repo_large.load_lora_weights("ehristoforu/dalle-3-xl-v2", adapter_name="base")
|
| 51 |
|
| 52 |
repo_customs = {
|
|
|
|
| 46 |
# repo_default.load_lora_weights("artificialguybr/PixelArtRedmond", adapter_name="pixel_base")
|
| 47 |
# repo_default.load_lora_weights("nerijs/pixel-art-xl", adapter_name="pixel_base_2")
|
| 48 |
|
| 49 |
+
repo_large = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16, use_safetensors=True, add_watermarker=False)
|
| 50 |
# repo_large.load_lora_weights("ehristoforu/dalle-3-xl-v2", adapter_name="base")
|
| 51 |
|
| 52 |
repo_customs = {
|