Spaces:
Running
on
Zero
Running
on
Zero
Update autoregressive/models/gpt_t2i.py
Browse files
autoregressive/models/gpt_t2i.py
CHANGED
|
@@ -450,7 +450,6 @@ class Transformer(nn.Module):
|
|
| 450 |
else:
|
| 451 |
freqs_cis = self.freqs_cis[input_pos]
|
| 452 |
# transformer blocks
|
| 453 |
-
print(input_pos)
|
| 454 |
for i, layer in enumerate(self.layers):
|
| 455 |
if i%self.layer_internal == 0:
|
| 456 |
if self.training:
|
|
|
|
| 450 |
else:
|
| 451 |
freqs_cis = self.freqs_cis[input_pos]
|
| 452 |
# transformer blocks
|
|
|
|
| 453 |
for i, layer in enumerate(self.layers):
|
| 454 |
if i%self.layer_internal == 0:
|
| 455 |
if self.training:
|