Spaces:
Running
on
T4
Running
on
T4
yuntian-deng
commited on
Commit
•
2a15275
1
Parent(s):
ba933fa
Update app.py
Browse files
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
|
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])
|