yuntian-deng commited on
Commit
6166e35
·
verified ·
1 Parent(s): ac347ce

Update app.py

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