omer11a commited on
Commit
c3855ce
·
1 Parent(s): 0f77b33

Removed xformers dependency

Browse files
Files changed (2) hide show
  1. app.py +0 -1
  2. requirements.txt +0 -1
app.py CHANGED
@@ -195,7 +195,6 @@ def main():
195
  scheduler = DDIMScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", clip_sample=False, set_alpha_to_one=False)
196
  model = StableDiffusionXLPipeline.from_pretrained(model_path, scheduler=scheduler, torch_dtype=torch.float16).to(device)
197
  model.unet.set_default_attn_processor()
198
- model.enable_xformers_memory_efficient_attention()
199
  model.enable_sequential_cpu_offload()
200
 
201
  nltk.download('averaged_perceptron_tagger')
 
195
  scheduler = DDIMScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", clip_sample=False, set_alpha_to_one=False)
196
  model = StableDiffusionXLPipeline.from_pretrained(model_path, scheduler=scheduler, torch_dtype=torch.float16).to(device)
197
  model.unet.set_default_attn_processor()
 
198
  model.enable_sequential_cpu_offload()
199
 
200
  nltk.download('averaged_perceptron_tagger')
requirements.txt CHANGED
@@ -19,5 +19,4 @@ torchmetrics==1.0.3
19
  torchvision==0.15.2
20
  tqdm==4.66.1
21
  transformers==4.32.0
22
- xformers==0.0.21
23
  spaces==0.24.2
 
19
  torchvision==0.15.2
20
  tqdm==4.66.1
21
  transformers==4.32.0
 
22
  spaces==0.24.2