rich-text-to-image

Runtime error

Songwei Ge commited on Mar 27, 2023

Commit

1ee21db

1 Parent(s): 61c1bd4

demo

Files changed (1) hide show

models/region_diffusion.py CHANGED Viewed

@@ -28,7 +28,7 @@ class RegionDiffusion(nn.Module):
         self.clip_gradient = False
         print(f'[INFO] loading stable diffusion...')
-        local_pretrained_dir = f'pretrained-guidance/v1'
         if not os.path.isdir(local_pretrained_dir):
             save_pretrained = True
             load_paths = 'runwayml/stable-diffusion-v1-5'

         self.clip_gradient = False
         print(f'[INFO] loading stable diffusion...')
+        local_pretrained_dir = "runwayml/stable-diffusion-v1-5"
         if not os.path.isdir(local_pretrained_dir):
             save_pretrained = True
             load_paths = 'runwayml/stable-diffusion-v1-5'