svjack commited on
Commit
65a22d6
1 Parent(s): 92b4dc2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -24,14 +24,14 @@ from diffusers import AutoPipelineForText2Image
24
  from diffusers.pipelines.wuerstchen.pipeline_wuerstchen_prior import DEFAULT_STAGE_C_TIMESTEPS
25
 
26
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
27
-
28
  assert os.path.exists("wuerstchen")
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",
33
  torch_dtype=torch.float32)
34
- '''
35
  pipe.to(device)
36
 
37
  pipe.safety_checker = None
 
24
  from diffusers.pipelines.wuerstchen.pipeline_wuerstchen_prior import DEFAULT_STAGE_C_TIMESTEPS
25
 
26
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
27
+ '''
28
  assert os.path.exists("wuerstchen")
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",
33
  torch_dtype=torch.float32)
34
+
35
  pipe.to(device)
36
 
37
  pipe.safety_checker = None