ImageInception
/

ArtifyAI-v1.1

StableDiffusionPipeline

Inference Endpoints

Model card Files Files and versions Community

Zaitounish commited on Oct 20, 2024

Commit

543ecf2

·

verified ·

1 Parent(s): 61c9a58

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -6,7 +6,7 @@ import torch
 # Load models
 t5_model = T5ForConditionalGeneration.from_pretrained('t5_model')
 t5_tokenizer = T5Tokenizer.from_pretrained('t5_tokenizer')
-ArtifyAI_model = StableDiffusionPipeline.from_pretrained('ArtifyAI_model', torch_dtype=torch.float16)
 ArtifyAI_model = ArtifyAI_model.to('cuda')
 # Combined pipeline

 # Load models
 t5_model = T5ForConditionalGeneration.from_pretrained('t5_model')
 t5_tokenizer = T5Tokenizer.from_pretrained('t5_tokenizer')
+ArtifyAI_model = StableDiffusionPipeline.from_pretrained('.', torch_dtype=torch.float16)
 ArtifyAI_model = ArtifyAI_model.to('cuda')
 # Combined pipeline