'init_image' -> 'image'

yyahav · Mar 22, 2023 · a07ab84 · a07ab84
1 parent c22c3ee
commit a07ab84
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vktrs/hf.py b/vktrs/hf.py
@@ -76,7 +76,7 @@ def get_image_for_prompt(
         prompt,
         **kwargs
     ):
-        f = self.text2img if kwargs.get('init_image') is None else self.img2img
+        f = self.text2img if kwargs.get('image') is None else self.img2img
         #if kwargs.get('image_consistency') is not None:
         #kwargs['strength'] = 1- kwargs['image_consistency'] 
         if kwargs.get('start_schedule') is not None: