Update app.py
Browse files
app.py
CHANGED
@@ -45,7 +45,7 @@ repo_nsfw_classifier = pipeline("image-classification", model="Falconsai/nsfw_im
|
|
45 |
|
46 |
repo_default = DiffusionPipeline.from_pretrained("fluently/Fluently-XL-Final", device=DEVICE, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False).to(DEVICE)
|
47 |
|
48 |
-
repo_neo = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", device=DEVICE, torch_dtype=torch.bfloat16, use_safetensors=
|
49 |
repo_neo.load_lora_weights("ehristoforu/dalle-3-xl-v2", adapter_name="base")
|
50 |
repo_neo.set_adapters(["base"], adapter_weights=[0.7])
|
51 |
|
|
|
45 |
|
46 |
repo_default = DiffusionPipeline.from_pretrained("fluently/Fluently-XL-Final", device=DEVICE, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False).to(DEVICE)
|
47 |
|
48 |
+
repo_neo = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", device=DEVICE, torch_dtype=torch.bfloat16, use_safetensors=False, add_watermarker=False, revision="refs/pr/1").to(DEVICE)
|
49 |
repo_neo.load_lora_weights("ehristoforu/dalle-3-xl-v2", adapter_name="base")
|
50 |
repo_neo.set_adapters(["base"], adapter_weights=[0.7])
|
51 |
|