UniPic

Running on Zero

yichenchenchen commited on 30 days ago

Commit

1e6ce87

verified ·

1 Parent(s): 5f34f32

Update inferencer.py

Files changed (1) hide show

inferencer.py CHANGED Viewed

@@ -131,7 +131,7 @@ class Inferencer:
         cfg_schedule="constant",
         temperature=1.0,
     ):
-        if not model:
             self.model = self.init_model()
         prompt = self.model.prompt_template["INSTRUCTION"].format(
             input=f"Generate an image: {raw_prompt.strip()}."

         cfg_schedule="constant",
         temperature=1.0,
     ):
+        if not self.model:
             self.model = self.init_model()
         prompt = self.model.prompt_template["INSTRUCTION"].format(
             input=f"Generate an image: {raw_prompt.strip()}."