Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -246,7 +246,7 @@ MAX_SEED = np.iinfo(np.int32).max
|
|
246 |
MAX_IMAGE_SIZE = 2048
|
247 |
|
248 |
if torch.cuda.is_available():
|
249 |
-
vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
|
250 |
token = os.environ.get("HF_TOKEN") # 从环境变量读取令牌
|
251 |
model_path = hf_hub_download(
|
252 |
repo_id="Menyu/MiaoMiao_Harem1.6", # 模型仓库名称(非完整URL)
|
@@ -255,7 +255,7 @@ if torch.cuda.is_available():
|
|
255 |
)
|
256 |
pipe = StableDiffusionXLPipeline.from_single_file(
|
257 |
model_path,
|
258 |
-
vae=vae,
|
259 |
use_safetensors=True,
|
260 |
torch_dtype=torch.float16,
|
261 |
)
|
|
|
246 |
MAX_IMAGE_SIZE = 2048
|
247 |
|
248 |
if torch.cuda.is_available():
|
249 |
+
#vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
|
250 |
token = os.environ.get("HF_TOKEN") # 从环境变量读取令牌
|
251 |
model_path = hf_hub_download(
|
252 |
repo_id="Menyu/MiaoMiao_Harem1.6", # 模型仓库名称(非完整URL)
|
|
|
255 |
)
|
256 |
pipe = StableDiffusionXLPipeline.from_single_file(
|
257 |
model_path,
|
258 |
+
#vae=vae, 内置VAE
|
259 |
use_safetensors=True,
|
260 |
torch_dtype=torch.float16,
|
261 |
)
|