Spaces:
Running
on
Zero
Running
on
Zero
rm cuda() and add variant=fp16
Browse files- scripts/process_utils.py +2 -2
scripts/process_utils.py
CHANGED
@@ -120,6 +120,7 @@ def initialize_refine_model():
|
|
120 |
sd_pipe = StableDiffusionPipeline.from_single_file(
|
121 |
refine_sd_model_path,
|
122 |
torch_dtype=torch_dtype,
|
|
|
123 |
use_safetensors=True
|
124 |
).to(device)
|
125 |
|
@@ -221,8 +222,7 @@ def generate_sotai_image(input_image: Image.Image, output_width: int, output_hei
|
|
221 |
|
222 |
# EasyNegativeV2の内容
|
223 |
easy_negative_v2 = "(worst quality, low quality, normal quality:1.4), lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, jpeg artifacts, signature, watermark, username, blurry, artist name, (bad_prompt_version2:0.8)"
|
224 |
-
|
225 |
-
sotai_gen_pipe = sotai_gen_pipe.cuda()
|
226 |
output = sotai_gen_pipe(
|
227 |
prompt,
|
228 |
image=[input_image, input_image],
|
|
|
120 |
sd_pipe = StableDiffusionPipeline.from_single_file(
|
121 |
refine_sd_model_path,
|
122 |
torch_dtype=torch_dtype,
|
123 |
+
variant="fp16",
|
124 |
use_safetensors=True
|
125 |
).to(device)
|
126 |
|
|
|
222 |
|
223 |
# EasyNegativeV2の内容
|
224 |
easy_negative_v2 = "(worst quality, low quality, normal quality:1.4), lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, jpeg artifacts, signature, watermark, username, blurry, artist name, (bad_prompt_version2:0.8)"
|
225 |
+
|
|
|
226 |
output = sotai_gen_pipe(
|
227 |
prompt,
|
228 |
image=[input_image, input_image],
|