Fix imports

pipeline.py

@@ -73,14 +73,10 @@ class CustomFlaxBartForConditionalGeneration(FlaxBartForConditionalGeneration):
 
 class PreTrainedPipeline():
     def __init__(self, path=""):
-        # IMPLEMENT_THIS
-        # Preload all the elements you are going to need at inference.
-        # For instance your model, processors, tokenizer that might be needed.
-        # This function is only called once, so do all the heavy processing I/O here"""
+        self.vqgan = VQModel.from_pretrained("flax-community/vqgan_f16_16384", revision="90cc46addd2dd8f5be21586a9a23e1b95aa506a9")
         self.tokenizer = BartTokenizer.from_pretrained(path)
         self.model = CustomFlaxBartForConditionalGeneration.from_pretrained(path)
 
-        self.vqgan = VQModel.from_pretrained("flax-community/vqgan_f16_16384", revision="90cc46addd2dd8f5be21586a9a23e1b95aa506a9")
 
  
     def __call__(self, inputs: str):

vqgan_jax/__init__.py

@@ -0,0 +1 @@
+from . import *