GonzaloMG commited on
Commit
01c23ae
1 Parent(s): 029bde2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -14,7 +14,7 @@ from huggingface_hub import hf_hub_download
14
  from Marigold.marigold import MarigoldPipeline
15
  from diffusers import AutoencoderKL, DDIMScheduler, UNet2DConditionModel
16
  from transformers import CLIPTextModel, CLIPTokenizer
17
- import xformers
18
 
19
  css = """
20
  #img-display-container {
@@ -48,10 +48,10 @@ pipe = MarigoldPipeline.from_pretrained(pretrained_model_name_or_path = checkpoi
48
  variant=variant,
49
  torch_dtype=dtype,
50
  )
51
- try:
52
- pipe.enable_xformers_memory_efficient_attention()
53
- except ImportError:
54
- pass # run without xformers
55
  pipe = pipe.to(DEVICE)
56
  pipe.unet.eval()
57
 
 
14
  from Marigold.marigold import MarigoldPipeline
15
  from diffusers import AutoencoderKL, DDIMScheduler, UNet2DConditionModel
16
  from transformers import CLIPTextModel, CLIPTokenizer
17
+ # import xformers
18
 
19
  css = """
20
  #img-display-container {
 
48
  variant=variant,
49
  torch_dtype=dtype,
50
  )
51
+ # try:
52
+ # pipe.enable_xformers_memory_efficient_attention()
53
+ # except ImportError:
54
+ # pass # run without xformers
55
  pipe = pipe.to(DEVICE)
56
  pipe.unet.eval()
57