update

model.py

@@ -125,6 +125,7 @@ class Model:
         print(W, H)
         self.t5_model.model.to('cuda')
         self.gpt_model_canny.to('cuda')
+        self.vq_model.to('cuda')
         print("after cuda", self.gpt_model_canny.adapter.model.embeddings.patch_embeddings.projection.weight)
 
         condition_img = self.get_control_canny(np.array(image), low_threshold,