Manjushri commited on
Commit
624960b
·
verified ·
1 Parent(s): 4f01bf2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -7
app.py CHANGED
@@ -39,13 +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
-
43
- if torch.cuda.is_available():
44
- torch_dtype = torch.bfloat16
45
- else:
46
- torch_dtype = torch.float32
47
-
48
- pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
49
  #pipe = pipe.to(device)
50
  pipe.enable_model_cpu_offload()
51
  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)
 
 
 
 
 
 
43
  #pipe = pipe.to(device)
44
  pipe.enable_model_cpu_offload()
45
  pipe.vae.enable_slicing()