Hugging Face's logo

Spaces:
ai-forever
/
slides_generator
Running on Zero

App Files Community
ai-forever commited on
Commit
46cddfc
·
verified ·
1 Parent(s): 2b6bf66

Update src/kandinsky.py

Browse files
Files changed (1) hide show
  1. src/kandinsky.py +1 -1
src/kandinsky.py CHANGED
@@ -29,7 +29,7 @@ if torch.cuda.is_available():
29
  else:
30
  torch_dtype = torch.float32
31
 
32
- pipe = DiffusionPipeline.from_pretrained(model_repo_id, variant="fp16", torch_dtype=torch_dtype)
33
  pipe = pipe.to(device)
34
 
35
  MAX_SEED = np.iinfo(np.int32).max
 
29
  else:
30
  torch_dtype = torch.float32
31
 
32
+ pipe = DiffusionPipeline.from_pretrained(model_repo_id, variant="fp16", torch_dtype=torch_dtype, force_download=True)
33
  pipe = pipe.to(device)
34
 
35
  MAX_SEED = np.iinfo(np.int32).max