Update modules/finetunedvqgan.py

modules/finetunedvqgan.py

@@ -31,4 +31,4 @@ class Generator:
         # Load the state dictionary into the models
         model_vaq.load_state_dict(checkpoint, strict=True)
 
-        return model_vaq
+        return model_vaq.eval()