Manjushri commited on
Commit
57f90b6
·
verified ·
1 Parent(s): 7da8eca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ def genie (Model, Prompt, negative_prompt, height, width, scale, steps, seed):
15
  if Model == "SD3":
16
  #torch.cuda.max_memory_allocated(device=device)
17
  torch.cuda.empty_cache()
18
- SD3 = QwenImagePipeline.from_pretrained("Qwen/Qwen-Image", torch_dtype=torch.bfloat16).to(device)
19
  torch.cuda.empty_cache()
20
  image=SD3(
21
  prompt=Prompt,
 
15
  if Model == "SD3":
16
  #torch.cuda.max_memory_allocated(device=device)
17
  torch.cuda.empty_cache()
18
+ SD3 = QwenImagePipeline.from_pretrained("Qwen/Qwen-Image", torch_dtype=torch.float16).to(device)
19
  torch.cuda.empty_cache()
20
  image=SD3(
21
  prompt=Prompt,