From 11b7740deeef7903e81ba4c65a45853323a5fd5e Mon Sep 17 00:00:00 2001 From: Volpeon Date: Thu, 22 Jun 2023 07:34:04 +0200 Subject: Remove training guidance_scale --- data/csv.py | 4 ---- 1 file changed, 4 deletions(-) (limited to 'data') diff --git a/data/csv.py b/data/csv.py index d726033..43bf14c 100644 --- a/data/csv.py +++ b/data/csv.py @@ -108,7 +108,6 @@ def collate_fn( dtype: torch.dtype, tokenizer: CLIPTokenizer, max_token_id_length: Optional[int], - with_guidance: bool, with_prior_preservation: bool, examples, ): @@ -195,7 +194,6 @@ class VlpnDataModule: tokenizer: CLIPTokenizer, constant_prompt_length: bool = False, class_subdir: str = "cls", - with_guidance: bool = False, num_class_images: int = 1, size: int = 768, num_buckets: int = 0, @@ -228,7 +226,6 @@ class VlpnDataModule: self.class_root.mkdir(parents=True, exist_ok=True) self.placeholder_tokens = placeholder_tokens self.num_class_images = num_class_images - self.with_guidance = with_guidance self.constant_prompt_length = constant_prompt_length self.max_token_id_length = None @@ -356,7 +353,6 @@ class VlpnDataModule: self.dtype, self.tokenizer, self.max_token_id_length, - self.with_guidance, self.num_class_images != 0, ) -- cgit v1.2.3-54-g00ecf