Spaces:
Sleeping
Sleeping
fix model name
Browse files
app.py
CHANGED
@@ -22,7 +22,7 @@ with torch.no_grad():
|
|
22 |
torch.cuda.empty_cache()
|
23 |
|
24 |
pipe = StableDiffusionPipeline.from_pretrained(
|
25 |
-
'raulm/
|
26 |
torch_dtype=torch.float16,
|
27 |
).to("cuda")
|
28 |
|
|
|
22 |
torch.cuda.empty_cache()
|
23 |
|
24 |
pipe = StableDiffusionPipeline.from_pretrained(
|
25 |
+
'raulm/hebru-small-batch-dreambooth',
|
26 |
torch_dtype=torch.float16,
|
27 |
).to("cuda")
|
28 |
|