Keshabwi66 commited on
Commit
fac4e49
·
1 Parent(s): c4c66c7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -118,6 +118,8 @@ unet_encoder.requires_grad_(False)
118
  text_encoder_one.requires_grad_(False)
119
  text_encoder_two.requires_grad_(False)
120
  unet_encoder.requires_grad_(False)
 
 
121
 
122
 
123
  pipe = TryonPipeline.from_pretrained(
 
118
  text_encoder_one.requires_grad_(False)
119
  text_encoder_two.requires_grad_(False)
120
  unet_encoder.requires_grad_(False)
121
+ unet.eval()
122
+ unet_encoder.eval()
123
 
124
 
125
  pipe = TryonPipeline.from_pretrained(