nupurkmr9 commited on
Commit
751f591
1 Parent(s): 3cf1717

Update trainer.py

Browse files
Files changed (1) hide show
  1. trainer.py +1 -1
trainer.py CHANGED
@@ -10,7 +10,7 @@ import gradio as gr
10
  import PIL.Image
11
  import torch
12
 
13
- os.environ['PYTHONPATH'] = f'lora:{os.getenv("PYTHONPATH", "")}'
14
 
15
 
16
  def pad_image(image: PIL.Image.Image) -> PIL.Image.Image:
 
10
  import PIL.Image
11
  import torch
12
 
13
+ os.environ['PYTHONPATH'] = f'custom-diffusion:{os.getenv("PYTHONPATH", "")}'
14
 
15
 
16
  def pad_image(image: PIL.Image.Image) -> PIL.Image.Image: