Manjushri commited on
Commit
9c1cd1d
1 Parent(s): d0d016e

Update app.py

Browse files

You need to remove the torch float16 from the model request.

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -52,7 +52,7 @@ current_model = models[1] if is_colab else models[0]
52
  current_model_path = current_model.path
53
 
54
  if is_colab:
55
- pipe = StableDiffusionPipeline.from_pretrained(current_model.path, torch_dtype=torch.float16, scheduler=scheduler, safety_checker=lambda images, clip_input: (images, False))
56
 
57
  else: # download all models
58
  print(f"{datetime.datetime.now()} Downloading vae...")
 
52
  current_model_path = current_model.path
53
 
54
  if is_colab:
55
+ pipe = StableDiffusionPipeline.from_pretrained(current_model.path, scheduler=scheduler, safety_checker=lambda images, clip_input: (images, False))
56
 
57
  else: # download all models
58
  print(f"{datetime.datetime.now()} Downloading vae...")