yichenchenchen commited on
Commit
b5c0902
·
verified ·
1 Parent(s): d094756

Update inferencer.py

Browse files
Files changed (1) hide show
  1. inferencer.py +2 -2
inferencer.py CHANGED
@@ -17,7 +17,7 @@ class UniPicV2Inferencer:
17
  self,
18
  model_path: str,
19
  qwen_vl_path: str,
20
- quant: str = "int4", # {"int4", "fp16"}
21
  image_size: int = 512,
22
  default_negative_prompt: str = "blurry, low quality, low resolution, distorted, deformed, broken content, missing parts, damaged details, artifacts, glitch, noise, pixelated, grainy, compression artifacts, bad composition, wrong proportion, incomplete editing, unfinished, unedited areas."
23
  ):
@@ -27,7 +27,7 @@ class UniPicV2Inferencer:
27
  self.image_size = image_size
28
  self.default_negative_prompt = default_negative_prompt
29
  self.device = torch.device("cuda")
30
- self.pipeline = None#self._init_pipeline()
31
 
32
  def _init_pipeline(self) -> StableDiffusion3KontextPipeline:
33
  print("Initializing UniPicV2 pipeline...")
 
17
  self,
18
  model_path: str,
19
  qwen_vl_path: str,
20
+ quant: str = "fp16", # {"int4", "fp16"}
21
  image_size: int = 512,
22
  default_negative_prompt: str = "blurry, low quality, low resolution, distorted, deformed, broken content, missing parts, damaged details, artifacts, glitch, noise, pixelated, grainy, compression artifacts, bad composition, wrong proportion, incomplete editing, unfinished, unedited areas."
23
  ):
 
27
  self.image_size = image_size
28
  self.default_negative_prompt = default_negative_prompt
29
  self.device = torch.device("cuda")
30
+ self.pipeline = None #self._init_pipeline()
31
 
32
  def _init_pipeline(self) -> StableDiffusion3KontextPipeline:
33
  print("Initializing UniPicV2 pipeline...")