Undo backend change to torch.float16

GeroVanMi · May 15, 2024 · 9d87d1f · 9d87d1f
1 parent 5787951
commit 9d87d1f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/backend/src/generate_images.py b/backend/src/generate_images.py
@@ -35,7 +35,7 @@ def initialize_pipeline(accelerator, config) -> DiffusionPipeline:
     pipeline = DDPMPipeline(
         unet=accelerator.unwrap_model(model), scheduler=noise_scheduler
     )
-    return pipeline.from_pretrained(config.output_dir, torch_dtype=torch.float16)
+    return pipeline.from_pretrained(config.output_dir)
 
 
 def upload_directory_with_transfer_manager(