Commit
•
a6b8e17
1
Parent(s):
6db545a
Update app.py
Browse files
app.py
CHANGED
@@ -6,7 +6,7 @@ import torch
|
|
6 |
from PIL import Image
|
7 |
import random
|
8 |
|
9 |
-
model_id = 'aipicasso/cool-japan-diffusion-2-1-2'
|
10 |
import transformers
|
11 |
transformers.utils.move_cache()
|
12 |
scheduler = EulerAncestralDiscreteScheduler(prediction_type="v_prediction") # .from_pretrained(model_id) # , subfolder="scheduler"
|
|
|
6 |
from PIL import Image
|
7 |
import random
|
8 |
|
9 |
+
model_id = 'https://huggingface.co/aipicasso/cool-japan-diffusion-2-1-2'
|
10 |
import transformers
|
11 |
transformers.utils.move_cache()
|
12 |
scheduler = EulerAncestralDiscreteScheduler(prediction_type="v_prediction") # .from_pretrained(model_id) # , subfolder="scheduler"
|