custom-diffusion

Build error

nupurkmr9 commited on Dec 16, 2022

Commit

751f591

1 Parent(s): 3cf1717

Update trainer.py

Files changed (1) hide show

trainer.py CHANGED Viewed

@@ -10,7 +10,7 @@ import gradio as gr
 import PIL.Image
 import torch
-os.environ['PYTHONPATH'] = f'lora:{os.getenv("PYTHONPATH", "")}'
 def pad_image(image: PIL.Image.Image) -> PIL.Image.Image:

 import PIL.Image
 import torch
+os.environ['PYTHONPATH'] = f'custom-diffusion:{os.getenv("PYTHONPATH", "")}'
 def pad_image(image: PIL.Image.Image) -> PIL.Image.Image: