Spaces:
Running
on
Zero
Running
on
Zero
Update inferencer.py
Browse files- 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 = "
|
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...")
|