Adding zerogpu attribute

model/model_manager.py

@@ -86,6 +86,14 @@ class ModelManager:
         if do_debug_packages:
             debug_packages()
 
+        try:
+            self.generate_image_ig.zerogpu = True
+            self.generate_image_ie.zerogpu = True
+            self.generate_video_vg.zerogpu = True
+        except:
+            print("Failed to set zerogpu for wrapper_fns")
+            pass
+
     def load_model_pipe(self, model_name):
         if not model_name in self.loaded_models:
             pipe = load_pipeline(model_name)