bariscal commited on
Commit
c60213b
·
1 Parent(s): 9adb7b6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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, 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):
 
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):