Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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",
|