Spaces:

mariarivaille
/

Diffusion_Models_Course_Space

Sleeping

Maria commited on Mar 10

Commit

2b7a847

1 Parent(s): 4f050b3

hw6

Files changed (1) hide show

infer.py CHANGED Viewed

@@ -125,10 +125,6 @@ def get_pipe(
         pipe.unet = PeftModel.from_pretrained(pipe.unet, unet_sub_dir, adapter_name=adapter_name)
         pipe.text_encoder = PeftModel.from_pretrained(pipe.text_encoder, text_encoder_sub_dir, adapter_name=adapter_name)
-        if torch_dtype == torch.float16:
-            pipe.unet.half()
-            pipe.text_encoder.half()
     return pipe

         pipe.unet = PeftModel.from_pretrained(pipe.unet, unet_sub_dir, adapter_name=adapter_name)
         pipe.text_encoder = PeftModel.from_pretrained(pipe.text_encoder, text_encoder_sub_dir, adapter_name=adapter_name)
     return pipe