parokshsaxena commited on
Commit
b065204
β€’
1 Parent(s): 2e6f7e8

remove enhanced garment net code

Browse files
Files changed (1) hide show
  1. src/enhanced_garment_net.py +4 -1
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
+ """