yuntian-deng commited on
Commit
2a15275
1 Parent(s): ba933fa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -36,7 +36,7 @@ def setup(device='cuda'):
36
  model_type = "EleutherAI/gpt-neo-125M"
37
  #encoder = AutoModel.from_pretrained(model_type).to(device)
38
  encoder = img_pipe.unet.text_encoder
39
- if True:
40
  l = len(img_pipe.unet.down_blocks)
41
  for i in range(l):
42
  img_pipe.unet.down_blocks[i] = torch.compile(img_pipe.unet.down_blocks[i])
 
36
  model_type = "EleutherAI/gpt-neo-125M"
37
  #encoder = AutoModel.from_pretrained(model_type).to(device)
38
  encoder = img_pipe.unet.text_encoder
39
+ if False:
40
  l = len(img_pipe.unet.down_blocks)
41
  for i in range(l):
42
  img_pipe.unet.down_blocks[i] = torch.compile(img_pipe.unet.down_blocks[i])