LoRA-DreamBooth-Training-UI

Runtime error

aellaboudy commited on May 3, 2023

Commit

bb53117

1 Parent(s): 470a7ca

Enable CPU prompts

Files changed (1) hide show

inference.py CHANGED Viewed

@@ -78,8 +78,8 @@ class InferencePipeline:
         n_steps: int,
         guidance_scale: float,
     ) -> PIL.Image.Image:
-        if not torch.cuda.is_available():
-            raise gr.Error('CUDA is not available.')
         self.load_pipe(lora_model_id)

         n_steps: int,
         guidance_scale: float,
     ) -> PIL.Image.Image:
+        #if not torch.cuda.is_available():
+        #   raise gr.Error('CUDA is not available.')
         self.load_pipe(lora_model_id)