Staticaliza commited on
Commit
fd2d1e0
·
verified ·
1 Parent(s): 15d96d1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=True, 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
 
 
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