Staticaliza commited on
Commit
b0433d2
·
verified ·
1 Parent(s): 30a95a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -4
app.py CHANGED
@@ -51,10 +51,7 @@ repo_default.load_lora_weights("nerijs/pixel-art-xl", adapter_name="pixel_base_2
51
  repo_pro = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16, use_safetensors=True)
52
  repo_pro.load_lora_weights(hf_hub_download("alimama-creative/FLUX.1-Turbo-Alpha", "diffusion_pytorch_model.safetensors"))
53
 
54
- repo_classic = StableDiffusionPipeline.from_pretrained("gsdf/Counterfeit-V2.5", revision="fp16", torch_dtype=torch.float16, use_safetensors=True, safety_checker=None)
55
- repo_classic.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
56
- repo_classic.enable_xformers_memory_efficient_attention()
57
- repo_classic = repo_classic.to("cuda")
58
 
59
  repo_customs = {
60
  "Default": repo_default,
 
51
  repo_pro = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16, use_safetensors=True)
52
  repo_pro.load_lora_weights(hf_hub_download("alimama-creative/FLUX.1-Turbo-Alpha", "diffusion_pytorch_model.safetensors"))
53
 
54
+ repo_classic = StableDiffusionPipeline.from_pretrained("gsdf/Counterfeit-V2.5", torch_dtype=torch.float16, use_safetensors=True, safety_checker=None)
 
 
 
55
 
56
  repo_customs = {
57
  "Default": repo_default,