Spaces:
Runtime error
Runtime error
Update app.py
Browse filesYou need to remove the torch float16 from the model request.
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,
|
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...")
|