update

.gitignore

@@ -1,3 +1,4 @@
 experiments/*
 trash/*
-scripts/*
+scripts/*
+__pycache__/*

inference.py

@@ -50,7 +50,7 @@ def inference_fn(
 
     # create inference pipeline
     device = 'cuda' if torch.cuda.is_available() else 'cpu'
-    pipe = StableDiffusionPipeline.from_pretrained(model_id,torch_dtype=torch.float16).to(device)
+    pipe = StableDiffusionPipeline.from_pretrained(os.path.join('experiments', model_id),torch_dtype=torch.float16).to(device)
 
     # make directory to save images
     image_root_folder = os.path.join('experiments', model_id, 'inference')