Hugging Face's logo

Spaces:
Shaleen123
/
kandinsky2.2
Runtime error

App Files Community
Shaleen123 commited on
Commit
eae528a
·
verified ·
1 Parent(s): 06e2634

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ if torch.cuda.is_available():
18
  pipe.enable_xformers_memory_efficient_attention()
19
  pipe = pipe.to(device)
20
  else:
21
- pipe = AutoPipelineForText2Image.from_pretrained("Shaleen123/kandinsky_2.5", use_safetensors=True, torch_dtype = torch.float16)
22
  pipe = pipe.to(device)
23
 
24
  MAX_SEED = np.iinfo(np.int32).max
 
18
  pipe.enable_xformers_memory_efficient_attention()
19
  pipe = pipe.to(device)
20
  else:
21
+ pipe = AutoPipelineForText2Image.from_pretrained("Shaleen123/kandinsky_2.5", use_safetensors=True)
22
  pipe = pipe.to(device)
23
 
24
  MAX_SEED = np.iinfo(np.int32).max