idle_tests

inference.py

@@ -47,7 +47,7 @@ class InferencePipeline:
             pipe = pipe.to(self.device)
 
         from src import diffuser_training
-        diffuser_training.load_model(pipe.text_encoder, pipe.tokenizer, pipe.unet, weight_path, compress=False)
+        diffuser_training.load_model(pipe.text_encoder, pipe.tokenizer, pipe.unet, weight_path, '<new1>')
 
         self.pipe = pipe
 

requirements.txt

@@ -6,6 +6,6 @@ Pillow==9.3.0
 torch==1.13.0
 torchvision==0.14.0
 transformers==4.25.1
-triton==2.0.0.dev20221030
+triton==2.0.0.dev20220701
 xformers==0.0.13
 clip_retrieval