KingNish commited on
Commit
a1318fe
Β·
verified Β·
1 Parent(s): ab51949

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -57,7 +57,6 @@ if torch.cuda.is_available():
57
  torch_dtype=torch.float16,
58
  use_safetensors=True,
59
  ).to("cuda")
60
- pipe.enable_xformers_memory_efficient_attention() # Enable memory efficient attention for less VRAM usage
61
  pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
62
  pipe.load_lora_weights("ehristoforu/dalle-3-xl-v2", weight_name="dalle-3-xl-lora-v2.safetensors", adapter_name="dalle")
63
  pipe.set_adapters("dalle")
 
57
  torch_dtype=torch.float16,
58
  use_safetensors=True,
59
  ).to("cuda")
 
60
  pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
61
  pipe.load_lora_weights("ehristoforu/dalle-3-xl-v2", weight_name="dalle-3-xl-lora-v2.safetensors", adapter_name="dalle")
62
  pipe.set_adapters("dalle")