对Stable Diffusion做fine-tune时遇见的bug

 由于我的显卡比较垃圾(3050),加载模型时都用的

torch.dtype=float16

 后来用下列方式保存模型:

pipeline = StableDiffusionPipeline(
        text_encoder=text_encoder,
        vae=vae,
        unet=unet,
        tokenizer=tokenizer,
        scheduler=PNDMScheduler(beta_start=0.00085,
                                beta_end=0.012,
                                beta_schedule='scaled_linear',
                                skip_prk_steps=True),
        safety_checker=StableDiffusionSafetyChecker.from_pretrained(
            'CompVis/stable-diffusion-safety-checker'),
        feature_extractor=CLIPFeatureExtractor.from_pretrained(
            'openai/clip-vit-base-patch32'),
    )
pipeline.save_pretrained('../modules/cat_toy')

报了错:

TypeError: Object of type dtype is not JSON serializable

在网上查到的各种方法都无法解决,折腾了一天后(想紫砂),又审视了一下这句话,意思是"dtype"这种类型不能JSON序列化,我的理解是,因为我的模型的dtype=float16,因此不能用这种方式保存。之后试了float32的模型,可以正常保存,但因为显存、内存太小,在本地跑不了😭😭😭

评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值