Spaces:
Running
on
Zero
Running
on
Zero
parokshsaxena
commited on
Commit
β’
b065204
1
Parent(s):
2e6f7e8
remove enhanced garment net code
Browse files
src/enhanced_garment_net.py
CHANGED
@@ -1,3 +1,4 @@
|
|
|
|
1 |
import torch
|
2 |
import torch.nn as nn
|
3 |
import torch.nn.functional as F
|
@@ -120,4 +121,6 @@ class EnhancedGarmentNetWithTimestep(nn.Module):
|
|
120 |
combined = comb_layer(torch.cat([feat, expanded_cond_emb], dim=1))
|
121 |
combined_features.append(combined)
|
122 |
|
123 |
-
return garment_out, combined_features
|
|
|
|
|
|
1 |
+
"""
|
2 |
import torch
|
3 |
import torch.nn as nn
|
4 |
import torch.nn.functional as F
|
|
|
121 |
combined = comb_layer(torch.cat([feat, expanded_cond_emb], dim=1))
|
122 |
combined_features.append(combined)
|
123 |
|
124 |
+
return garment_out, combined_features
|
125 |
+
|
126 |
+
"""
|