Manjushri commited on
Commit
b36176f
·
verified ·
1 Parent(s): 97ddc59

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -39,7 +39,7 @@ def genie (Model, Prompt, negative_prompt, height, width, scale, steps, seed):
39
  progress=gr.Progress(track_tqdm=True)
40
  torch.cuda.empty_cache()
41
  model_repo_id = "stabilityai/stable-diffusion-3.5-large-turbo"
42
- pipe = DiffusionPipeline.from_pretrained(model_repo_id)
43
  pipe = pipe.to(device)
44
  pipe.enable_model_cpu_offload()
45
  pipe.vae.enable_slicing()
 
39
  progress=gr.Progress(track_tqdm=True)
40
  torch.cuda.empty_cache()
41
  model_repo_id = "stabilityai/stable-diffusion-3.5-large-turbo"
42
+ pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype = torch.float32)
43
  pipe = pipe.to(device)
44
  pipe.enable_model_cpu_offload()
45
  pipe.vae.enable_slicing()