alexnasa commited on
Commit
cee33b2
·
verified ·
1 Parent(s): 8838fa7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -70,11 +70,8 @@ text_encoder.requires_grad_(False)
70
  unet.requires_grad_(False)
71
  controlnet.requires_grad_(False)
72
 
73
- if is_xformers_available():
74
- unet.enable_xformers_memory_efficient_attention()
75
- controlnet.enable_xformers_memory_efficient_attention()
76
- else:
77
- raise ValueError("xformers is not available. Make sure it is installed correctly")
78
 
79
  # Get the validation pipeline
80
  validation_pipeline = StableDiffusionControlNetPipeline(
 
70
  unet.requires_grad_(False)
71
  controlnet.requires_grad_(False)
72
 
73
+ unet.enable_xformers_memory_efficient_attention()
74
+ controlnet.enable_xformers_memory_efficient_attention()
 
 
 
75
 
76
  # Get the validation pipeline
77
  validation_pipeline = StableDiffusionControlNetPipeline(