Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -4,7 +4,7 @@ from diffusers import StableDiffusionPipeline
|
|
| 4 |
|
| 5 |
|
| 6 |
#gr.Interface.load("models/bariscal/cbst_style")
|
| 7 |
-
pipe = StableDiffusionPipeline.from_pretrained("bariscal/cbst_style", safety_checker=None
|
| 8 |
|
| 9 |
|
| 10 |
def inference(prompt, negative_prompt, num_samples, height=512, width=512, num_inference_steps=50, guidance_scale=7.5):
|
|
|
|
| 4 |
|
| 5 |
|
| 6 |
#gr.Interface.load("models/bariscal/cbst_style")
|
| 7 |
+
pipe = StableDiffusionPipeline.from_pretrained("bariscal/cbst_style", safety_checker=None) #, torch_dtype=torch.float16
|
| 8 |
|
| 9 |
|
| 10 |
def inference(prompt, negative_prompt, num_samples, height=512, width=512, num_inference_steps=50, guidance_scale=7.5):
|