OpenShape
/

openshape-demo-support

Model card Files Files and versions Community

eliphatfs commited on May 29, 2023

Commit

4dbdaec

·

1 Parent(s): 44a6c30

Fix.

Files changed (1) hide show

openshape/demo/sd_pc2img.py +1 -1

openshape/demo/sd_pc2img.py CHANGED Viewed

@@ -16,7 +16,7 @@ class Wrapper(transformers.modeling_utils.PreTrainedModel):
 half = torch.float16 if torch.cuda.is_available() else torch.bfloat16
 pipe = StableUnCLIPImg2ImgPipeline.from_pretrained(
     "diffusers/stable-diffusion-2-1-unclip-i2i-l",
-    torch_dtype=half, variation="fp16",
     image_encoder = Wrapper()
 )
 if torch.cuda.is_available():

 half = torch.float16 if torch.cuda.is_available() else torch.bfloat16
 pipe = StableUnCLIPImg2ImgPipeline.from_pretrained(
     "diffusers/stable-diffusion-2-1-unclip-i2i-l",
+    torch_dtype=half, variant="fp16",
     image_encoder = Wrapper()
 )
 if torch.cuda.is_available():