svjack commited on
Commit
3dbac1e
·
1 Parent(s): b136c13

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ from diffusers.pipelines.wuerstchen.pipeline_wuerstchen_prior import DEFAULT_STA
26
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
27
 
28
 
29
- pipe = AutoPipelineForText2Image.from_pretrained(Path("wuerstchen"),
30
  torch_dtype=torch.float32)
31
  '''
32
  pipe = AutoPipelineForText2Image.from_pretrained("warp-ai/wuerstchen",
 
26
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
27
 
28
 
29
+ pipe = AutoPipelineForText2Image.from_pretrained(Path("wuerstchen"), local_files_only = True,
30
  torch_dtype=torch.float32)
31
  '''
32
  pipe = AutoPipelineForText2Image.from_pretrained("warp-ai/wuerstchen",