Daniton commited on
Commit
1a3784b
Β·
1 Parent(s): f7ea2e2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ local_dir = "./model/"
9
  pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, cache_dir=local_dir)
10
 
11
  # Load the model from the local directory on subsequent runs
12
- pipe = StableDiffusionPipeline.from_pretrained(local_dir, torch_dtype=torch.float16)
13
 
14
  def generate_image(prompt):
15
  image = pipe(prompt).images[0]
 
9
  pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, cache_dir=local_dir)
10
 
11
  # Load the model from the local directory on subsequent runs
12
+ #pipe = StableDiffusionPipeline.from_pretrained(local_dir, torch_dtype=torch.float16)
13
 
14
  def generate_image(prompt):
15
  image = pipe(prompt).images[0]